Home
last modified time | relevance | path

Searched refs:TFE_TensorHandleCopyToDevice (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_remote_test_util.cc120 TFE_TensorHandleCopyToDevice(h_task0, ctx, task2_name, status)); in TestRemoteExecuteSilentCopies()
125 TFE_TensorHandleCopyToDevice(h1_task0, ctx, task2_name, status); in TestRemoteExecuteSilentCopies()
190 auto* retval_task0 = TFE_TensorHandleCopyToDevice( in TestRemoteExecuteSilentCopies()
Dc_api_remote_test.cc71 TFE_TensorHandleCopyToDevice(h0_task0, ctx, remote_device_name, status); in TestRemoteExecute()
74 TFE_TensorHandleCopyToDevice(h1_task0, ctx, remote_device_name, status); in TestRemoteExecute()
Dc_api_experimental_test.cc385 TFE_TensorHandle* copy = TFE_TensorHandleCopyToDevice(m, ctx, name, status); in TEST()
396 TFE_TensorHandleCopyToDevice(copy_aliased, ctx, "CPU:0", status); in TEST()
451 TFE_TensorHandle* hgpu = TFE_TensorHandleCopyToDevice( in TEST()
500 TFE_TensorHandle* h_cpu = TFE_TensorHandleCopyToDevice( in TEST()
Dc_api_cluster_test.cc73 TFE_TensorHandleCopyToDevice(retvals[0], ctx, local_device_name, status); in CheckRemoteMatMulExecutesOK()
163 TFE_TensorHandleCopyToDevice(h0_task0_new, ctx, remote_device_name, status); in TestRemoteExecuteChangeServerDef()
172 auto* h0_task1_new = TFE_TensorHandleCopyToDevice( in TestRemoteExecuteChangeServerDef()
Dc_api_test.cc227 TFE_TensorHandleCopyToDevice(hcpu, ctx, name.c_str(), status.get()); in TensorHandleCopyBetweenDevices()
234 TFE_TensorHandleCopyToDevice(hdevice, ctx, name.c_str(), status.get()); in TensorHandleCopyBetweenDevices()
242 TFE_TensorHandleCopyToDevice(hdevice2, ctx, kCPUDevice, status.get()); in TensorHandleCopyBetweenDevices()
288 TFE_TensorHandleCopyToDevice(hcpu, ctx, kErrorDevice, status.get()); in TensorHandleCopyBetweenDevicesError()
296 TFE_TensorHandleCopyToDevice(hcpu, ctx, kCPUDevice, status.get()); in TensorHandleCopyBetweenDevicesError()
360 TFE_TensorHandleCopyToDevice(hcpu, ctx, gpu_1_name.c_str(), status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
363 TFE_TensorHandle* hdevice2 = TFE_TensorHandleCopyToDevice( in TensorHandleCopyBetweenTwoGPUDevices()
369 TFE_TensorHandleCopyToDevice(hdevice2, ctx, kCPUDevice, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
419 TFE_TensorHandle* hgpu = TFE_TensorHandleCopyToDevice( in TensorHandleSilentCopy()
569 TFE_TensorHandle* hgpu = TFE_TensorHandleCopyToDevice( in TEST()
[all …]
Dcustom_device_test.cc45 TFE_TensorHandleCopyToDevice(hcpu, context, name, status.get()); in TEST()
284 TFE_TensorHandleCopyToDevice(hcpu.get(), context.get(), custom0, in TEST()
291 TFE_TensorHandleCopyToDevice(hcpu.get(), context.get(), custom1, in TEST()
Dc_api_distributed_test.cc587 TFE_TensorHandleCopyToDevice(h0_task0, ctx, remote_device_name, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
590 TFE_TensorHandleCopyToDevice(h1_task0, ctx, remote_device_name, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
Dcustom_device_testutil.cc71 TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice( in CopyToLoggingDevice()
Dc_api.h192 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopyToDevice(
Dc_api.cc906 TFE_TensorHandle* TFE_TensorHandleCopyToDevice(TFE_TensorHandle* h, in TFE_TensorHandleCopyToDevice() function
/external/tensorflow/tensorflow/go/
Dtensor_handle.go164 newTh := C.TFE_TensorHandleCopyToDevice(th.c, c.c, n, status.c)
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc319 handle = make_safe(TFE_TensorHandleCopyToDevice(handle.get(), ctx, in ConvertToEagerTensorUncached()
620 TFE_TensorHandle* handle = TFE_TensorHandleCopyToDevice( in EagerTensor_copy_to_device()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_test.cc132 TensorHandlePtr device_value(TFE_TensorHandleCopyToDevice( in TEST()
161 TensorHandlePtr copy_back(TFE_TensorHandleCopyToDevice( in TEST()
Dparallel_device_lib.cc277 TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice( in CopyToParallelDevice()