Home
last modified time | relevance | path

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

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