Searched refs:TF_SetDevice (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| D | c_api_session_coordination_test.cc | 104 TF_SetDevice(set_desc, task1_name); in TEST_P()
114 TF_SetDevice(get_desc, task2_name); in TEST_P()
|
| /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 | 258 TF_CAPI_EXPORT extern void TF_SetDevice(TF_OperationDescription* desc,
|
| D | c_api_function_test.cc | 223 TF_SetDevice(desc, "/cpu:0"); in UseHelper()
|
| D | c_api.cc | 789 void TF_SetDevice(TF_OperationDescription* desc, const char* device) { in TF_SetDevice() function
|
| /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/go/ |
| D | graph.go | 357 C.TF_SetDevice(cdesc, cdevice)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_wrapper.cc | 1127 m.def("TF_SetDevice", TF_SetDevice); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1941 pywrap_tf_session.TF_SetDevice(op_desc, compat.as_str(node_def.device))
|