Searched refs:TFE_OpSetDevice (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | c_api_test.cc | 171 TFE_OpSetDevice(matmul, remote_device_name, status); in TestRemoteExecute() 253 TFE_OpSetDevice(matmul, task1_name, status); in TestRemoteExecuteSilentCopies() 324 TFE_OpSetDevice(matmul, remote_device_name, status); in CheckRemoteMatMulExecutesOK() 674 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy() 720 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopyLocal() 755 TFE_OpSetDevice(matmul, "GPU:0", status); in SetAndGetOpDevices() 760 TFE_OpSetDevice(matmul, "CPU:0", status); in SetAndGetOpDevices() 843 TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get()); in TEST() 920 TFE_OpSetDevice(matmul, "/job:localhost/replica:0/task:0/device:CPU:0", in Execute_MatMul_CPU_Runtime_Error() 924 TFE_OpSetDevice(matmul2, "/job:localhost/replica:0/task:0/device:CPU:0", in Execute_MatMul_CPU_Runtime_Error()
|
D | c_api_experimental_test.cc | 59 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status); in ExecuteWithProfiling()
|
D | c_api.h | 273 TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
|
D | c_api.cc | 595 void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { in TFE_OpSetDevice() function
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tensor.cc | 189 TFE_OpSetDevice(op, device_name, out_status); in EagerCast() local
|
D | pywrap_tfe_src.cc | 720 TFE_OpSetDevice(op, device_name, out_status); in TFE_Py_Execute() 2228 TFE_OpSetDevice(op, parent_op_exec_info.device_name, status); 2621 TFE_OpSetDevice(op, op_exec_info.device_name, status);
|
/external/tensorflow/tensorflow/c/ |
D | c_api_experimental.cc | 8574 TFE_OpSetDevice(queueOp.get(), DEFAULT_CPU_DEVICE, status); in createTFEQueue() 8606 TFE_OpSetDevice(op, DEFAULT_CPU_DEVICE, status); in createTFEEnqueue() 8628 TFE_OpSetDevice(op, DEFAULT_CPU_DEVICE, status); in createTFEDequeue()
|