Home
last modified time | relevance | path

Searched refs:CopyTensorHandleToDevice (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc98 ImmediateExecutionTensorHandle* CustomDeviceOpHandler::CopyTensorHandleToDevice( in CopyTensorHandleToDevice() function in tensorflow::CustomDeviceOpHandler
128 return context->CopyTensorHandleToDevice(handle, device_name, status); in CopyTensorHandleToDevice()
Dcustom_device_op_handler.h39 ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
Dcustom_device.cc68 context_->GetCustomDeviceOpHandler().CopyTensorHandleToDevice( in Resolve()
Dcore.cc106 ImmediateExecutionTensorHandle* EagerContext::CopyTensorHandleToDevice( in CopyTensorHandleToDevice() function in tensorflow::EagerContext
Dcontext.h133 ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_context.h95 virtual ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
Dc_api.cc898 unwrapped_ctx->GetCustomDeviceOpHandler().CopyTensorHandleToDevice( in TFE_TensorHandleCopyToDevice()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h111 tensorflow::ImmediateExecutionTensorHandle* CopyTensorHandleToDevice(
Dc_api_tfrt.cc996 ContextInterface::CopyTensorHandleToDevice( in CopyTensorHandleToDevice() function in tfrt::tf::ContextInterface