- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for set_requested_device (0.39 sec)
-
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); Node* call; TF_ASSERT_OK(MakeXlaCompiledKernel(root.graph(), "cluster_0", "C", &call)); call->AddAttr(kXlaHasReferenceVarsAttr, false); call->set_requested_device(kXlaDeviceName); Node* write_op = MakeWrite(root, "write"); write_op->AddAttr(kXlaHasReferenceVarsAttr, false); root.graph()->AddControlEdge(call, write_op); std::unique_ptr<Graph> graph;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc
auto arg6 = ops::_Arg(scope.WithOpName("w_0_arg"), DT_RESOURCE, 6); auto add_attrs = [](Node* node) { node->AddAttr(kXlaClusterIdAttr, "launch0"); node->set_requested_device("/gpu:0"); }; auto b_identity = ops::Identity(scope.WithOpName("B_identity"), arg1); add_attrs(b_identity.node()); auto read_u = ops::ReadVariableOp(scope.WithOpName("ReadU"), arg4, DT_FLOAT);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 14.7K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} void TF_SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { using tensorflow::RecordMutation; mutex_lock l(graph->mu); op->node.set_requested_device(device); RecordMutation(graph, *op, "setting device"); } void TF_RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { mutex_lock l(graph->mu);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/python_api.cc
TF_ClearAttr(graph, op, attr_name, status); } void SetFullType(TF_Graph* graph, TF_Operation* op, const TF_Buffer* full_type_proto) { TF_SetFullType(graph, op, full_type_proto); } void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { TF_SetRequestedDevice(graph, op, device); } void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, TF_Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 12 18:48:56 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/c/python_api.h
TF_Status* status); // Sets the experimental_type` field in the node_def Protocol Buffer. void SetFullType(TF_Graph* graph, TF_Operation* op, const TF_Buffer* full_type_proto); void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device); // Updates 'dst' to consume 'new_src'. void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, TF_Status* status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 12 18:48:56 UTC 2023 - 3.5K bytes - Viewed (0)