Home
last modified time | relevance | path

Searched refs:src_device_context (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtensor_util.cc81 tensorflow::DeviceContext* src_device_context = nullptr; in TransferTensorToDevice() local
83 src_device_context = in TransferTensorToDevice()
106 "copy", src_device_context, dst_device_context, srcd, dstd, in TransferTensorToDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dtensor_handle.cc1004 tensorflow::DeviceContext* src_device_context = nullptr; in CopyToDevice() local
1006 src_device_context = in CopyToDevice()
1024 tensorflow::CopyTensor::ViaDMA("copy", src_device_context, dst_device_context, in CopyToDevice()