Searched refs:CopyTensorToDevice (Results 1 – 5 of 5) sorted by relevance
38 virtual Status CopyTensorToDevice(
106 *status = dev->CopyTensorToDevice(handle, &result); in CopyTensorHandleToDevice()
38 Status CopyTensorToDevice(ImmediateExecutionTensorHandle* tensor, in CopyTensorToDevice() function in tensorflow::eager::__anon325885840111::TestCustomDevice
252 Tensor CopyTensorToDevice(Device* device, const Tensor& tensor) { in CopyTensorToDevice() function290 input_buffer = CopyTensorToDevice(device, *input); in RunCollective()
422 tensorflow::Status CopyTensorToDevice( in CopyTensorToDevice() function in tensorflow::__anon9d665ed60211::CustomDeviceAPI