Searched refs:CopyToParallelDevice (Results 1 – 3 of 3) sorted by relevance
155 parallel_device.CopyToParallelDevice( in ExecuteWithSpecialOps()236 TFE_TensorHandle* CopyToParallelDevice(TFE_Context* context, in CopyToParallelDevice() function243 dev.CopyToParallelDevice(context, tensor, status)); in CopyToParallelDevice()368 device->copy_tensor_to_device = &CopyToParallelDevice; in AllocateParallelDevice()
67 std::unique_ptr<ParallelTensor> CopyToParallelDevice(TFE_Context* context,
272 std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( in CopyToParallelDevice() function in tensorflow::parallel_device::ParallelDevice