| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/ |
| H A D | conversion.cc | 49 const KernelFallbackTensor& tensor, const tfrt::Device& src, in ConvertKernelFallbackTensorToStringHostTensor()
82 static tfrt::AsyncValueRef<KernelFallbackTensor>
88 KernelFallbackTensor(tensor.shape(), tensor.dtype(), tf_tensor); in ConvertStringHostTensorToKernelFallbackTensor()
94 const KernelFallbackTensor& tensor, const tfrt::Device& src, in ConvertKernelFallbackTensorToDenseHostTensor()
118 static tfrt::AsyncValueRef<KernelFallbackTensor>
124 KernelFallbackTensor src_knfb_tensor(tensor.shape(), tensor.dtype(), in ConvertDenseHostTensorToKernelFallbackTensor()
129 static tfrt::AsyncValueRef<KernelFallbackTensor> TransferKernelFallback( in TransferKernelFallback()
130 const KernelFallbackTensor& tensor, const tfrt::Device& src, in TransferKernelFallback()
|
| H A D | conversion.h | 33 class KernelFallbackTensor; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_tensor.h | 44 class KernelFallbackTensor final
46 public tfrt::TensorTraits<KernelFallbackTensor> {
48 explicit KernelFallbackTensor(::tensorflow::Tensor tensor) in KernelFallbackTensor() function
50 KernelFallbackTensor(const tfrt::TensorShape& shape, tfrt::DType dtype, in KernelFallbackTensor() function
54 static KernelFallbackTensor Create(const tensorflow::Tensor& tensor) { in Create()
55 return KernelFallbackTensor(tensor); in Create()
|
| H A D | tensor_util.cc | 29 tfrt::AsyncValueRef<KernelFallbackTensor> TransferTensorToDevice( in TransferTensorToDevice()
30 const tfrt::ExecutionContext& exec_ctx, const KernelFallbackTensor& tensor, in TransferTensorToDevice()
45 return tfrt::MakeAvailableAsyncValueRef<KernelFallbackTensor>(*src); in TransferTensorToDevice()
74 return tfrt::MakeAvailableAsyncValueRef<KernelFallbackTensor>(dst); in TransferTensorToDevice()
77 auto result = tfrt::MakeUnconstructedAsyncValueRef<KernelFallbackTensor>(); in TransferTensorToDevice()
|
| H A D | tensor_util.h | 25 class KernelFallbackTensor; variable
29 tfrt::AsyncValueRef<KernelFallbackTensor> TransferTensorToDevice(
30 const tfrt::ExecutionContext& exec_ctx, const KernelFallbackTensor& tensor,
|
| H A D | kernel_fallback_execute.cc | 39 using tensorflow::KernelFallbackTensor;
117 outputs[i]->emplace<KernelFallbackTensor>( in KernelFallbackExecute()
118 KernelFallbackTensor::Create(op_kernel_context.output(i))); in KernelFallbackExecute()
|
| H A D | kernel_fallback_kernels.cc | 110 cpu_op_handler->AddImplicitConversion(KernelFallbackTensor::kTensorType, in AddKernelFallbackImplicitConversionKernel()
112 cpu_op_handler->AddImplicitConversion(KernelFallbackTensor::kTensorType, in AddKernelFallbackImplicitConversionKernel()
114 cpu_op_handler->AddImplicitConversion(KernelFallbackTensor::kTensorType, in AddKernelFallbackImplicitConversionKernel()
|
| H A D | kernel_fallback_op_handler.cc | 264 invocation.exec_ctx, KernelFallbackTensor::kTensorType); in MakeOp()
|
| H A D | kernel_fallback_execute_compat.cc | 381 KernelFallbackExecuteCompatAsyncInternal<KernelFallbackTensor>( in KernelFallbackExecuteCompatCoreRuntimeDispatch()
384 KernelFallbackExecuteCompatSyncInternal<KernelFallbackTensor>( in KernelFallbackExecuteCompatCoreRuntimeDispatch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
| H A D | conversion.cc | 35 const KernelFallbackTensor &tensor, const tfrt::Device &src, in ConvertKernelFallbackToRuntimeFallbackTensor()
62 static tfrt::AsyncValueRef<KernelFallbackTensor>
72 KernelFallbackTensor(tensor.shape(), tensor.dtype(), *tf_tensor); in ConvertRuntimeFallbackToKernelFallbackTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/gpu/ |
| H A D | conversion_function.cc | 44 const KernelFallbackTensor& tensor, const tfrt::gpu::GpuDevice& src, in ConvertKernelFallbackTensorOnGpuToDenseGpuTensor()
77 static llvm::Expected<KernelFallbackTensor>
93 return KernelFallbackTensor{tensor.shape(), tensor.dtype(), in ConvertDenseGpuTensorToKernelFallbackTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/ |
| H A D | runtime.cc | 98 tensorflow::KernelFallbackTensor::kTensorType, in InitializeOpHandlers()
101 tensorflow::KernelFallbackTensor::kTensorType, in InitializeOpHandlers()
104 tensorflow::KernelFallbackTensor::kTensorType, in InitializeOpHandlers()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | tensor_util.cc | 115 if (auto* knfbt = llvm::dyn_cast<tensorflow::KernelFallbackTensor>(&tensor)) { in TFRTTensorToTFTensor()
136 auto knfbt = MakeAvailableAsyncValueRef<tensorflow::KernelFallbackTensor>( in TFTensorToTFRTTensorHandle()
183 llvm::dyn_cast<tensorflow::KernelFallbackTensor>(&host_tensor)) { in CreateTFTensorFromTensorHandle()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_kernels.cc | 858 KernelFallbackTensor::kTensorType); in CoreRTTensorHandleToFallbackTensorInternal()
1008 tfrt::MakeAvailableAsyncValueRef<KernelFallbackTensor>(tf_tensor); in FallbackTensorToCoreRTTensorHandleInternal()
|