- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 155 for created (0.73 sec)
-
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/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
} auto dataset_while = builder.create<TF::WhileRegionOp>( reduce_dataset.getLoc(), while_input_types, /*input=*/while_input_values, /*parallel_iterations=*/10, false, /*shape_invariant=*/false); // `_lower_using_switch_merge` is the default for While ops created // in TensorFlow and allows lowering to V1 control flow for loop // parallelization.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K 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/c/eager/c_api_experimental_test.cc
TF_DeleteStatus(status); } // 6. Read `var` using `ctx_1`. This read should succeed since `ctx_1` was // created with `isolate_session_state` set to false, and update should // preserve it. { // Create a handle to `var`, using `ctx_1`. TFE_TensorHandle* var_handle = CreateVarHandle(ctx_1, remote_device, /*variable_name=*/"var");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc
moved_ops_list.push_back(new_if_op.getOperation()); } } } // Create the Yield ops for both branches with merged results. // `builder` is the OpBuilder. // `if_op_segment` refers to the segment of IfRegions to be merged. // `return_indices` refers to the return indices to be kept in merged IfRegion // `new_if_op` refers to the created new IfRegion void CreateYieldOps(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables_test.cc
std::string tmp_dir{}; if (!env_->LocalTempFilename(&tmp_dir)) { return absl::InternalError("Failed to create temp file."); } TF_CHECK_OK(env_->CreateDir(tmp_dir)); return tmp_dir; } // Parses `module_op_str` to create a `ModuleOp`. Checks whether the created // module op is valid. mlir::OwningOpRef<mlir::ModuleOp> ParseModuleOpString( const absl::string_view module_op_str) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 16.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
absl::StrCat("device: ", DEVICE_TPU_SYSTEM, " device")); devices->push_back(std::make_unique<VirtualDevice>(options.env, attrs)); VLOG(1) << "Created TPU_SYSTEM device. This host has " << device_count << " TPUs"; return absl::OkStatus(); } } // namespace void RegisterTpuDeviceToDeviceCopy() { static auto* const register_tpu_tpu_copy = new CopyTensor::Registration(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
namespace mlir { namespace TF { namespace { // Creates ConstOp for int32_t value. ConstOp createI32ConstOp(int32_t value, Location loc, PatternRewriter* rewriter) { auto int_attr = IntegerAttr::get(rewriter->getIntegerType(32), value); return rewriter->create<ConstOp>(loc, int_attr); } // Creates ConstantOp for array of int32_t.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
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)