- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 235 for created (0.4 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
builder, block.getParent()->getLoc(), execute, device); builder->create<tf_device::ReturnOp>(block.getParent()->getLoc(), block_launch_op.getResults()); } return success(); } // Creates a `tf.TPUCompileSucceededAssert` operation that parses compilation // status of `compile_op` to check whether compilation is successful.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
func::FuncOp parent_func, ModuleOp module, const std::string& name, bool flag_for_inlining) { // Moves all of the Operations in 'ops' into a newly created func.FuncOp // function named 'name' and replaces the original ops with a call to the // newly created function using a tf.StatefulPartitionedCall. Here, // 'parent_func' is the function that holds the original set of ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
ModuleOp module) { SymbolTableCollection table; SymbolUserMap symbol_map(table, module); llvm::DenseMap<func::FuncOp, llvm::DenseSet<func::FuncOp>> caller_callee_map; // Creates work queue for determining reachability below. std::queue<func::FuncOp> function_worklist; for (auto func : module.getOps<func::FuncOp>()) { for (auto user : symbol_map.getUsers(func)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/debugging/dump_tensor_op.cc
} // `DumpTensor` op saves entire value of input to as a tensor proto into a // specified directory and filename. When enabled is set to false, op is // disabled and won't save any value. It also creates `QuantizationUnit` proto // with `func_name` and `node_name` to identify the op. REGISTER_OP("DumpTensor") .Input("tensor_data: T") .Attr("log_dir_path: string") .Attr("file_name: string") .Attr("T: type")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 03:12:17 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
module_op.emitError() << "quant-convert-tpu-model-to-cpu pattern " "conversion did not converge."; signalPassFailure(); return; } } } // namespace // Creates an instance of `ConvertTpuModelToCpuPass`. std::unique_ptr<OperationPass<ModuleOp>> CreateConvertTpuModelToCpuPass() { return std::make_unique<ConvertTpuModelToCpuPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
#include "tensorflow/compiler/jit/xla_compilation_cache.pb.h" #include "xla/service/hlo.pb.h" #include "tensorflow/core/platform/path.h" #include "tensorflow/core/public/session.h" namespace tensorflow { namespace { // Creates a float tensor of linearly increasing values, starting from offset. Tensor CreateInputTensor(const TensorShape& shape, float offset) { Tensor tensor(DT_FLOAT, shape); for (int64 i = 0; i < tensor.flat<float>().size(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto reshape_sizes = DenseIntElementsAttr::get(reshape_type, new_shape); auto reshape_value = builder->create<TF::ConstOp>(input.getLoc(), reshape_sizes); return builder->create<TF::ReshapeOp>(input.getLoc(), reshape_result_type, input, reshape_value); } // Creates transpose op for shape to depth transform. TF::TransposeOp GetTransposeOpForConv2DFilter(OpBuilder* builder, Value input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc
ctx); if (failed(applyPatternsAndFoldGreedily(func_op, std::move(patterns_2)))) { signalPassFailure(); } } } } // namespace // Creates an instance of the TensorFlow dialect PrepareQuantize pass. std::unique_ptr<OperationPass<ModuleOp>> CreatePrepareQuantizePass( const bool enable_per_channel_quantized_weight, const int bit_width) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 05:11:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc
if (failed(ApplyQuantizationSpec(spec, module_op))) { signalPassFailure(); return; } } } } // namespace // Creates `LiftQuantizableSpotsAsFunctionsPass` with user-defined // `QuantizationSpecs`. std::unique_ptr<OperationPass<ModuleOp>> CreateLiftQuantizableSpotsAsFunctionsPass( const QuantizationSpecs& quantization_specs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
} else { if (IsCompatibleTypeWithTFLCastOp(value.getType()) && IsCompatibleTypeWithTFLCastOp(type)) { auto cast = b.create<CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } else { auto cast = b.create<TF::CastOp>(yield_op->getLoc(), type, value); args.push_back(cast); } } } args.append(new_args.begin(), new_args.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0)