- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 155 for created (0.19 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
if (tpu_replicated_input.getIsMirroredVariable()) { mirrored_variable_indices.push_back(pos_and_input.index()); } } // Create replicate op. auto result_types = GetClusterResultTypes(cluster, partitioned_outputs); auto replicate_op = builder.create<mlir::tf_device::ReplicateOp>( cluster.getLoc(), num_replicas, llvm::SmallDenseMap<llvm::StringRef,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc
// Create new ConstantOp-Dequantize-Operation sequences. At this moment, // old ConstantOp is guaranteed to have one F32->F16 cast regardless of // its number of users. rewriter.setInsertionPointAfter(op); auto new_const = rewriter.create<arith::ConstantOp>( op->getLoc(), new_result_type, new_value_attr); auto dq = rewriter.create<DQ>(op->getLoc(), old_result_type, new_const);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
func::FuncOp parent_func, ModuleOp module, const std::string& name) { // 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 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
} SmallVector<Value> return_values; for (Value result : results) { return_values.push_back(mapping.lookupOrNull(result)); } builder.create<func::ReturnOp>(location, return_values); // Create a function call to the newly created function. StringAttr new_func_name = InsertToSymbolTable(*module, *wrap_func, func_name); builder.setInsertionPointAfter(result_op); ValueRange new_results =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
SliceInputs slice_inputs_int64 = MakeSliceIndexAndSizeInt64(host_scope, slice_inputs); // Create a list of all control dependencies to be copied when possibly // replacing nodes related to slice_size. Node* old_size; std::vector<const Edge*> old_size_ctrl_deps; TF_RETURN_IF_ERROR(slice->input_node(2, &old_size));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
Value tensor_i32 = builder.create<TF::CastOp>( loc, tensor_type.clone(builder.getIntegerType(32)), tensor); auto reduced = builder.create<TF::SumOp>(loc, tensor_i32, reduction_indices_value, /*keep_dims=*/builder.getBoolAttr(true)); auto mul_op = builder.create<TF::MulOp>(loc, zp, reduced);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
return failure(); } // Compute multiplier = scale / sqrt(variance + epsilon) Value multiplier = rewriter.create<mhlo::AddOp>( bn_op.getLoc(), bn_op.getVariance(), epsilon); multiplier = rewriter.create<mhlo::RsqrtOp>(bn_op.getLoc(), multiplier); multiplier = rewriter.create<mhlo::MulOp>(bn_op.getLoc(), multiplier, bn_op.getScale());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc
// be created). static void DumpModule(mlir::ModuleOp module, std::string file_prefix) { std::string prefix = GetDumpDirFromEnvVar(); if (prefix.empty()) return; auto* env = tensorflow::Env::Default(); auto status = env->RecursivelyCreateDir(prefix); if (!status.ok()) { LOG(WARNING) << "cannot create directory '" << prefix << "': " << status.message();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 18.5K bytes - Viewed (0) -
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/c/c_api_experimental.cc
TF_Output output{dequeue_op, 0}; TF_Tensor* ret; TF_SessionRun(session, /*run_options*/ nullptr, // input related parameters /*inputs*/ nullptr, /*input_values*/ nullptr, /*ninputs*/ 0, // output related parameters /*outputs*/ &output, /*output_values*/ &ret, /*noutputs*/ 1, /*targets*/ nullptr, /*ntargets*/ 0,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0)