Searched refs:CopyToParallelDevice (Results 1 – 3 of 3) sorted by relevance
163 parallel_device.CopyToParallelDevice( in ExecuteWithSpecialOps()266 TFE_TensorHandle* CopyToParallelDevice(TFE_Context* context, in CopyToParallelDevice() function411 device->copy_tensor_to_device = &CopyToParallelDevice; in AllocateParallelDevice()
68 std::unique_ptr<ParallelTensor> CopyToParallelDevice(TFE_Context* context,
298 std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( in CopyToParallelDevice() function in tensorflow::parallel_device::ParallelDevice