Home
last modified time | relevance | path

Searched refs:TF_SetDevice (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_test_util.cc213 TF_SetDevice(desc, op_device.c_str()); in BinaryOpHelper()
299 TF_SetDevice(desc, "/cpu:0"); in Split3Helper()
Dc_api.h466 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
Dc_api_function_test.cc223 TF_SetDevice(desc, "/cpu:0"); in UseHelper()
1701 TF_SetDevice(desc, "/device:XLA_CPU:0"); in TEST_F()
Dc_api.cc1211 void TF_SetDevice(TF_OperationDescription* desc, const char* device) { in TF_SetDevice() function
/external/tensorflow/tensorflow/go/
Dgraph.go334 C.TF_SetDevice(cdesc, cdevice)
/external/tensorflow/tensorflow/java/src/main/native/
Doperation_builder_jni.cc140 TF_SetDevice(d, cdevice); in Java_org_tensorflow_OperationBuilder_setDevice()
/external/tensorflow/tensorflow/python/framework/
Dops.py1798 c_api.TF_SetDevice(op_desc, compat.as_str(node_def.device))