Home
last modified time | relevance | path

Searched refs:MoveHostBufferToTfTensor (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_tensor.cc175 tensorflow::Tensor tensor = MoveHostBufferToTfTensor( in MoveDHTToRuntimeFallbackTensor()
190 tensorflow::Tensor tensor = MoveHostBufferToTfTensor( in CopyRefDHTToRuntimeFallbackTensor()
H A Druntime_fallback_kernels.cc1204 MoveHostBufferToTfTensor(dht->buffer(), dht->dtype(), dht->shape()); in ConvertTFRTTensorToTFTensorHandle()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dtensor_util.h41 tensorflow::Tensor MoveHostBufferToTfTensor(
H A Dtensor_util.cc31 tensorflow::Tensor MoveHostBufferToTfTensor(RCReference<HostBuffer> host_buffer, in MoveHostBufferToTfTensor() function
H A Dtensor_util_test.cc55 MoveHostBufferToTfTensor(dht.ReleaseBuffer(), dht.dtype(), dht.shape()); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/
H A Dconversion.cc123 MoveHostBufferToTfTensor(tensor.buffer(), tensor.dtype(), tensor.shape()); in ConvertDenseHostTensorToKernelFallbackTensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtensor_util.cc121 return tensorflow::tfd::MoveHostBufferToTfTensor( in TFRTTensorToTFTensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.cc893 tensorflow::tfd::MoveHostBufferToTfTensor(dht->buffer(), dht->dtype(), in TFTensorHandleFromInterface()