Searched refs:CopyTensorToDevice (Results 1 – 7 of 7) sorted by relevance
262 Tensor CopyTensorToDevice(Device* device, const Tensor& tensor) { in CopyTensorToDevice() function300 input_buffer = CopyTensorToDevice(device, *input); in RunCollective()
38 virtual Status CopyTensorToDevice(
106 *status = dev->CopyTensorToDevice(handle, &result); in CopyTensorHandleToDevice()
38 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice() function in tensorflow::eager::__anonb9701f0e0111::TestCustomDevice
131 LazyTensorPtr CopyTensorToDevice(const BackendDevice& device);
311 LazyTensorPtr LazyTensor::CopyTensorToDevice(const BackendDevice& device) { in CopyTensorToDevice() function in torch::lazy::LazyTensor
424 tensorflow::Status CopyTensorToDevice( in CopyTensorToDevice() function in tensorflow::__anon816881a00211::CustomDeviceAPI