Home
last modified time | relevance | path

Searched refs:copy_tensor_from_device (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc167 custom_device.copy_tensor_from_device = &CopyTensorFromLoggingDevice; in RegisterLoggingDevice()
191 custom_device->copy_tensor_from_device = &CopyTensorFromLoggingDevice; in AllocateLoggingDevice()
Dc_api_experimental.h482 TFE_TensorHandle* (*copy_tensor_from_device)(TFE_Context* context, member
Dc_api.cc445 TFE_TensorHandle* result_handle = device_.copy_tensor_from_device( in CopyTensorFromDevice()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device.cc412 device->copy_tensor_from_device = &CopyTensorFromParallelDevice; in AllocateParallelDevice()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc2016 device->copy_tensor_from_device = &CopyFromDTensorDevice; in AllocateDTensorDevice()