Searched refs:TFE_OpSetDevice (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_remote_test.cc | 88 TFE_OpSetDevice(matmul, remote_device_name, status); in TestRemoteExecute()
172 TFE_OpSetDevice(matmul, task1_name, status); in TestRemoteExecuteSilentCopies()
261 TFE_OpSetDevice(matmul, remote_device_name, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
316 TFE_OpSetDevice(matmul, remote_device_name, status); in CheckRemoteMatMulExecutesOK()
|
| D | c_api_test.cc | 388 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopy()
436 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status.get()); in TensorHandleSilentCopyLocal()
473 TFE_OpSetDevice(matmul, "GPU:0", status); in SetAndGetOpDevices()
478 TFE_OpSetDevice(matmul, "CPU:0", status); in SetAndGetOpDevices()
561 TFE_OpSetDevice(shape_op, gpu_device_name.c_str(), status.get()); in TEST()
640 TFE_OpSetDevice(matmul, "/job:localhost/replica:0/task:0/device:CPU:0", in Execute_MatMul_CPU_Runtime_Error()
644 TFE_OpSetDevice(matmul2, "/job:localhost/replica:0/task:0/device:CPU:0", in Execute_MatMul_CPU_Runtime_Error()
|
| D | c_api.h | 254 TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
|
| D | c_api_experimental_test.cc | 59 TFE_OpSetDevice(matmul, gpu_device_name.c_str(), status); in ExecuteWithProfiling()
|
| D | c_api.cc | 1220 void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { in TFE_OpSetDevice() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | eval_util.cc | 112 TFE_OpSetDevice(op, node_def->device().c_str(), status); in EvaluateOperation()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 353 TFE_OpSetDevice(queueOp.get(), DEFAULT_CPU_DEVICE, status); in createTFEQueue()
385 TFE_OpSetDevice(op, DEFAULT_CPU_DEVICE, status); in createTFEEnqueue()
407 TFE_OpSetDevice(op, DEFAULT_CPU_DEVICE, status); in createTFEDequeue()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_operation_builder_jni.cc | 122 TFE_OpSetDevice(op, cname, status); in Java_org_tensorflow_EagerOperationBuilder_setDevice()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 212 TFE_OpSetDevice(op, device_name, out_status); in EagerCast() local
|
| D | pywrap_tfe_src.cc | 3244 TFE_OpSetDevice(op, parent_op_exec_info.device_name, status);
|