Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc171 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()
Dc_api_experimental_test.cc59 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status); in ExecuteWithProfiling()
Dc_api.h273 TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
Dc_api.cc595 void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { in TFE_OpSetDevice() function
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc189 TFE_OpSetDevice(op, device_name, out_status); in EagerCast() local
Dpywrap_tfe_src.cc720 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/
Dc_api_experimental.cc8574 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()