Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc164 TFE_TensorHandleCopyToDevice(h0_task0, ctx, remote_device_name, status); in TestRemoteExecute()
167 TFE_TensorHandleCopyToDevice(h1_task0, ctx, remote_device_name, status); in TestRemoteExecute()
248 TFE_TensorHandleCopyToDevice(h1_task0, ctx, task2_name, status); in TestRemoteExecuteSilentCopies()
261 auto* retval_task0 = TFE_TensorHandleCopyToDevice( in TestRemoteExecuteSilentCopies()
333 TFE_TensorHandleCopyToDevice(retvals[0], ctx, local_device_name, status); in CheckRemoteMatMulExecutesOK()
404 TFE_TensorHandleCopyToDevice(h0_task0_new, ctx, remote_device_name, status); in TestRemoteExecuteChangeServerDef()
413 auto* h0_task1_new = TFE_TensorHandleCopyToDevice( in TestRemoteExecuteChangeServerDef()
486 TFE_TensorHandleCopyToDevice(hcpu, ctx, name.c_str(), status.get()); in TensorHandleCopyBetweenDevices()
493 TFE_TensorHandleCopyToDevice(hdevice, ctx, name.c_str(), status.get()); in TensorHandleCopyBetweenDevices()
501 TFE_TensorHandleCopyToDevice(hdevice2, ctx, kCPUDevice, status.get()); in TensorHandleCopyBetweenDevices()
[all …]
Dc_api.h211 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_TensorHandleCopyToDevice(
Dc_api.cc825 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.cc97 TFE_TensorHandleCopyToDevice(handle, context, device, status.get()); in CopyToDevice()
Dpywrap_tfe_src.cc2201 *handle = tensorflow::make_safe(TFE_TensorHandleCopyToDevice(