- Sort Score
- Num 10 results
- Language All
Results 1 - 5 of 5 for TF_SetDevice (0.07 seconds)
-
tensorflow/c/c_test_util.cc
// TODO(iga): Convert all these helpers and tests to use floats because // they are usually available on GPUs. After doing this, remove TF_SetDevice // call in c_api_function_test.cc TF_SetDevice(desc, "/cpu:0"); *op = TF_FinishOperation(desc, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); ASSERT_NE(*op, nullptr); }
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 04 05:55:32 GMT 2025 - 17.8K bytes - Click Count (1) -
tensorflow/c/c_api.h
TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperation( TF_Graph* graph, const char* op_type, const char* oper_name); // Specify the device for `desc`. Defaults to empty, meaning unconstrained. TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc, const char* device); // The calls to TF_AddInput and TF_AddInputList must match (in number,
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Click Count (0) -
tensorflow/c/c_api_function_test.cc
for (auto input : inputs) { TF_AddInput(desc, input); } // Set device to CPU because some ops inside the function might not be // available on GPU. TF_SetDevice(desc, "/cpu:0"); *op = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); ASSERT_NE(*op, nullptr); } FunctionDef fdef() { tensorflow::FunctionDef fdef;
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Wed Jan 07 04:56:09 GMT 2026 - 63.9K bytes - Click Count (1) -
tensorflow/c/c_api.cc
const char* oper_name) { mutex_lock l(graph->mu); return TF_NewOperationLocked(graph, op_type, oper_name); } void TF_SetDevice(TF_OperationDescription* desc, const char* device) { desc->node_builder.Device(device); } void TF_AddInput(TF_OperationDescription* desc, TF_Output input) {Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Oct 04 05:55:32 GMT 2025 - 102.4K bytes - Click Count (0) -
tensorflow/c/c_api_test.cc
TF_TString_Copy(&data[i], source.c_str(), source.length()); } TF_DeleteTensor(t); } } // namespace } // namespace tensorflow // TODO(josh11b): Test: // * TF_SetDevice(desc, "/job:worker"); // * control inputs / outputs // * targets
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Wed Jan 07 04:56:09 GMT 2026 - 97.3K bytes - Click Count (0)