- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 52 for set_device (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/fake_session.cc
t.scalar<tensorflow::ResourceHandle>()().set_device(kDeviceName); outputs->push_back(t); } else if (output_name == "var2") { Tensor t = Tensor(tensorflow::DT_RESOURCE, tensorflow::TensorShape({1})); t.scalar<tensorflow::ResourceHandle>()().set_name("var2"); t.scalar<tensorflow::ResourceHandle>()().set_device(kDeviceName); outputs->push_back(t);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_variable_loading_test.cc
auto& test_params = GetParam(); DataType dtype = std::get<0>(test_params); TensorShape shape(std::get<1>(test_params)); SavedVariable saved_variable; saved_variable.set_dtype(dtype); saved_variable.set_device("/job:localhost/replica:0/task:0/device:CPU:1"), shape.AsProto(saved_variable.mutable_shape()); std::unique_ptr<Variable> var; TF_ASSERT_OK(internal::LoadSavedVariable(context(), saved_variable, &var));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
} }; void TestBitcastOp(Tensor* input_tensor, DataType out_type, TensorShape expected_shape, error::Code expected_code) { Status status; NodeDef def; def.set_op("Bitcast"); def.set_device(DEVICE_CPU); AttrValue typeAttr; SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr); (*def.mutable_attr())["T"] = typeAttr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/c/kernels/summary_op_test.cc
error::Code expected_code) { // Initialize node used to fetch OpKernel Status status; NodeDef def; def.set_op("ScalarSummary"); def.set_device(DEVICE_CPU); AttrValue valuesTypeAttr; SetAttrValue(values->dtype(), &valuesTypeAttr); (*def.mutable_attr())["T"] = valuesTypeAttr; def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype())));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
Var* var = CreateVariable<T>(name, shape, data); ResourceMgr* rm = device_->resource_manager(); TF_ASSERT_OK(rm->Create(rm->default_container(), name, var)); ResourceHandle handle; handle.set_device(device_->name()); handle.set_container(rm->default_container()); handle.set_name(name); TypeIndex type_index = TypeIndex::Make<Var>(); handle.set_hash_code(type_index.hash_code());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
const char* node_name, Status* status) { NodeDef def; def.set_op(op_name); def.set_name(node_name); def.set_device(device_name); def.add_input("input1"); def.add_input("input2"); AttrValue v; v.set_type(DataType::DT_FLOAT); (*def.mutable_attr())["SomeDataTypeAttr"] = v;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
// Target the XLA CPU/GPU backends. VLOG(2) << "Replacing with XlaLaunch"; VLOG(2) << "Device is " << launch->requested_device(); def.set_op("XlaLaunch"); def.set_device(launch->requested_device()); AddNodeAttr("Tconstants", DataTypeVector{}, &def); AddNodeAttr("Targs", arg_types, &def); AddNodeAttr("Nresources", num_variables, &def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
// associated NodeDef is not updated. for (int i = 0, e = inst->getNumOperands(); i < e; ++i) { node_def->add_input(); } if (auto attr = inst->getAttrOfType<mlir::StringAttr>("device")) { node_def->set_device(std::string(attr.getValue())); } // Add the node debug info. TF_RETURN_IF_ERROR(ConvertLocation( inst->getLoc(), name, node_def->mutable_experimental_debug_info())); return node_def; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
Var* var = CreateVariable<T>(name, shape, data); ResourceMgr* rm = device_->resource_manager(); TF_ASSERT_OK(rm->Create(rm->default_container(), name, var)); ResourceHandle handle; handle.set_device(device_->name()); handle.set_container(rm->default_container()); handle.set_name(name); TypeIndex type_index = TypeIndex::Make<Var>(); handle.set_hash_code(type_index.hash_code());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// name_in is copied here because name may be modified below if // rewrite_subgraph_fn is true. string name = name_in; call_node_def_.set_op(name); call_node_def_.set_name(name); call_node_def_.set_device(device_); if (rewrite_subgraph_fn) { std::vector<OutputTensor> arg_source_tensors(args_by_src_.size()); for (const auto& arg : args_by_src_) { arg_source_tensors.at(arg.second) = arg.first; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0)