| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_tensor.cc | 60 RuntimeFallbackTensor::RuntimeFallbackTensor(const TensorShape& shape, in RuntimeFallbackTensor() function in tensorflow::tfd::RuntimeFallbackTensor
100 void RuntimeFallbackTensor::Print(tfrt::raw_ostream& os) const { in Print()
146 tfrt::Expected<RuntimeFallbackTensor>
168 return RuntimeFallbackTensor(shape, GetTfrtDtype(dtype), std::move(owned_th)); in CreateRuntimeFallbackTensorFromTfTensorHandle()
171 RuntimeFallbackTensor MoveDHTToRuntimeFallbackTensor(DenseHostTensor&& dht, in MoveDHTToRuntimeFallbackTensor()
182 return RuntimeFallbackTensor(dht.shape(), dht.dtype(), in MoveDHTToRuntimeFallbackTensor()
186 RuntimeFallbackTensor CopyRefDHTToRuntimeFallbackTensor( in CopyRefDHTToRuntimeFallbackTensor()
196 return RuntimeFallbackTensor(dht.shape(), dht.dtype(), in CopyRefDHTToRuntimeFallbackTensor()
200 RuntimeFallbackTensor CopySHTToRuntimeFallbackTensor( in CopySHTToRuntimeFallbackTensor()
206 return RuntimeFallbackTensor(sht.shape(), sht.dtype(), in CopySHTToRuntimeFallbackTensor()
|
| H A D | conversion_function.cc | 41 const RuntimeFallbackTensor &tensor, const tfrt::CpuDevice &src, in ConvertRuntimeFallbackTensorToDenseHostTensor()
66 const RuntimeFallbackTensor &tensor, const tfrt::Device &src, in ConvertRuntimeFallbackTensorToStringHostTensor()
94 static tfrt::AsyncValueRef<RuntimeFallbackTensor>
108 return tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in ConvertScalarHostTensorToRuntimeFallbackTensor()
112 static tfrt::AsyncValueRef<RuntimeFallbackTensor>
119 return tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in ConvertDenseHostTensorToRuntimeFallbackTensor()
123 static tfrt::AsyncValueRef<RuntimeFallbackTensor>
129 return tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in ConvertStringHostTensorToRuntimeFallbackTensor()
133 static tfrt::Expected<RuntimeFallbackTensor>
134 TransferRuntimeFallbackToAnotherDevice(const RuntimeFallbackTensor &tensor, in TransferRuntimeFallbackToAnotherDevice()
|
| H A D | runtime_fallback_tensor.h | 32 class RuntimeFallbackTensor final
34 public tfrt::TensorTraits<RuntimeFallbackTensor> {
36 explicit RuntimeFallbackTensor(const tfrt::TensorShape& shape,
65 tfrt::Expected<RuntimeFallbackTensor>
69 RuntimeFallbackTensor MoveDHTToRuntimeFallbackTensor(
72 RuntimeFallbackTensor CopyRefDHTToRuntimeFallbackTensor(
75 RuntimeFallbackTensor CopySHTToRuntimeFallbackTensor(
|
| H A D | runtime_fallback_op_handler.cc | 120 const RuntimeFallbackTensor& result_tensor, in GetDeviceFromFallbackTensor()
183 result_tensor_av.get<RuntimeFallbackTensor>(), exec_ctx); in GetResultDevice()
204 result_tensor_av_ref.get<RuntimeFallbackTensor>(), exec_ctx); in GetResultDevice()
206 result_tensor_av_ref.get<RuntimeFallbackTensor>(), exec_ctx)); in GetResultDevice()
242 invocation.exec_ctx, RuntimeFallbackTensor::kTensorType); in MakeOp()
270 result_tensor_av->get<RuntimeFallbackTensor>().GetTensorHandle(); in MakeOp()
|
| H A D | runtime_fallback_kernels.cc | 137 static AsyncValueRef<RuntimeFallbackTensor> CreateRuntimeFallbackTensor( in CreateRuntimeFallbackTensor()
160 return tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in CreateRuntimeFallbackTensor()
171 static std::pair<RuntimeFallbackTensor, Chain> TfdMoveDHTToTFT( in TfdMoveDHTToTFT()
185 static void TfdConvertTFTToDHT(Argument<RuntimeFallbackTensor> tft, in TfdConvertTFTToDHT()
203 static void TfdPrintTFT(Argument<RuntimeFallbackTensor> tft, in TfdPrintTFT()
603 llvm::cast<RuntimeFallbackTensor>(input_tensor)->GetTensorHandle()); in RuntimeFallbackExecute()
642 results[i] = tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in RuntimeFallbackExecute()
713 input_tensor_av->get<RuntimeFallbackTensor>().GetTensorHandle(); in RuntimeFallbackKernel()
799 tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in RuntimeFallbackKernel()
1125 llvm::SmallVector<RuntimeFallbackTensor, 4> tfrt_tensor_args; in RuntimeFallbackExecuteOp()
[all …]
|
| H A D | conversion_function.h | 34 class RuntimeFallbackTensor; variable
38 const RuntimeFallbackTensor &tensor, const tfrt::CpuDevice &src,
|
| H A D | runtime_fallback_batch_tf_opkernels.cc | 72 if (auto* rtfbt = llvm::dyn_cast<RuntimeFallbackTensor>(&tensor)) { in ConvertTFRTTensorToTFTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| H A D | conversion_function.cc | 44 static tfrt::Expected<RuntimeFallbackTensor>
57 return RuntimeFallbackTensor(gpu_tensor.shape(), gpu_tensor.dtype(), in CopyRefGpuTensorToRuntimeFallbackTensor()
68 const RuntimeFallbackTensor& tensor, const tfrt::Device& src, in ConvertRuntimeFallbackTensorToDenseGpuTensor()
189 static tfrt::AsyncValueRef<RuntimeFallbackTensor>
221 return tfrt::MakeAvailableAsyncValueRef<RuntimeFallbackTensor>( in ConvertDenseGpuTensorToRuntimeFallbackTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| H A D | cpu_registration.cc | 53 tensorflow::tfd::RuntimeFallbackTensor::kTensorType, in RegisterCpuOpHandler()
56 tensorflow::tfd::RuntimeFallbackTensor::kTensorType, in RegisterCpuOpHandler()
59 tensorflow::tfd::RuntimeFallbackTensor::kTensorType, in RegisterCpuOpHandler()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
| H A D | conversion.cc | 34 static RuntimeFallbackTensor ConvertKernelFallbackToRuntimeFallbackTensor( in ConvertKernelFallbackToRuntimeFallbackTensor()
58 return RuntimeFallbackTensor(tensor.shape(), tensor.dtype(), in ConvertKernelFallbackToRuntimeFallbackTensor()
64 const RuntimeFallbackTensor &tensor, const tfrt::Device &src, in ConvertRuntimeFallbackToKernelFallbackTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/ |
| H A D | runtime.cc | 89 tensorflow::tfd::RuntimeFallbackTensor::kTensorType, in InitializeOpHandlers()
92 tensorflow::tfd::RuntimeFallbackTensor::kTensorType, in InitializeOpHandlers()
95 tensorflow::tfd::RuntimeFallbackTensor::kTensorType, in InitializeOpHandlers()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 298 assert(llvm::isa<tensorflow::tfd::RuntimeFallbackTensor>(tensor_.get())); in Type()
299 return tensor_.get<tensorflow::tfd::RuntimeFallbackTensor>() in Type()
394 assert(async_tensor->IsType<tensorflow::tfd::RuntimeFallbackTensor>()); in DataType()
395 return async_tensor->get<tensorflow::tfd::RuntimeFallbackTensor>() in DataType()
792 tensorflow::tfd::RuntimeFallbackTensor>( in CreateLocalHandle()
820 tensorflow::tfd::RuntimeFallbackTensor>( in CreateLocalHandle()
832 MakeAvailableAsyncValueRef<tensorflow::tfd::RuntimeFallbackTensor>( in CreateLocalHandle()
858 tfrt::Expected<tensorflow::tfd::RuntimeFallbackTensor> in CreateLocalHandleFromTFTensor()
867 MakeAvailableAsyncValueRef<tensorflow::tfd::RuntimeFallbackTensor>( in CreateLocalHandleFromTFTensor()
888 llvm::dyn_cast<tensorflow::tfd::RuntimeFallbackTensor>(&tensor)) in TFTensorHandleFromInterface()
|