- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 130 for created (1.6 sec)
-
tensorflow/c/while_loop_test.cc
} // This is a basic test to make sure the C++ gradient code can handle while // loops created by the C API (which calls the C++ API under the hood). There // are more while loop gradient tests in cc/framework/while_gradients_test.cc. TEST_F(CApiWhileLoopTest, Gradients) { Init(1); // Create loop: while (i < 10) i += 1 TF_Operation* ten = ScalarConst(10, params_->cond_graph, s_); TF_Operation* less_than =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 11 06:05:56 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
if (only_one_return_value) { return_types.resize(1); } auto type = FunctionType::get(region.getContext(), input_types, return_types); // Create new function and extract region body into the function. auto outlined_func = builder.create<func::FuncOp>(loc, name, type); Region& func_region = outlined_func.getBody(); func_region.takeBody(region); Block& first_block = func_region.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K 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/mlir/lite/transforms/legalize_tensorlist.cc
if (subtypes.size() != 1) { return std::nullopt; } return subtypes[0].getElementType(); } // Create an `ConstBytesAttr` which encodes the options // for the `tf.custom` tensor list op to be created. If the given // op is not a `tf.TensorList*` op, return empty, although this case // should never be trigged in practice since patterns are only applied // on `tf.TensorList*` ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
xla_tensor->AllocateShapedBuffer(device_tensor->dtype(), shape, client_, stream_->parent()->device_ordinal())); // The cpu_tensor and literal that we created here hold the data of host // tensor in descending layout. The layout could be different from layout in // device_tensor (but the logical shape has to be the same). The
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K 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/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/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)