Searched refs:TF_SetDevice (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 213 TF_SetDevice(desc, op_device.c_str()); in BinaryOpHelper()
299 TF_SetDevice(desc, "/cpu:0"); in Split3Helper()
|
| D | c_api.h | 257 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
|
| D | c_api_function_test.cc | 223 TF_SetDevice(desc, "/cpu:0"); in UseHelper()
1738 TF_SetDevice(desc, "/device:XLA_CPU:0"); in TEST_F()
|
| D | c_api.cc | 787 void TF_SetDevice(TF_OperationDescription* desc, const char* device) { in TF_SetDevice() function
|
| /external/tensorflow/tensorflow/go/ |
| D | graph.go | 327 C.TF_SetDevice(cdesc, cdevice)
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_operation_builder_jni.cc | 142 TF_SetDevice(d, cdevice); in Java_org_tensorflow_GraphOperationBuilder_setDevice()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_wrapper.cc | 1084 m.def("TF_SetDevice", TF_SetDevice); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1641 pywrap_tf_session.TF_SetDevice(op_desc, compat.as_str(node_def.device))
|