Home
last modified time | relevance | path

Searched refs:CopyToParallelDevice (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.cc163 parallel_device.CopyToParallelDevice( in ExecuteWithSpecialOps()
266 TFE_TensorHandle* CopyToParallelDevice(TFE_Context* context, in CopyToParallelDevice() function
411 device->copy_tensor_to_device = &CopyToParallelDevice; in AllocateParallelDevice()
Dparallel_device_lib.h68 std::unique_ptr<ParallelTensor> CopyToParallelDevice(TFE_Context* context,
Dparallel_device_lib.cc298 std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( in CopyToParallelDevice() function in tensorflow::parallel_device::ParallelDevice