Home
last modified time | relevance | path

Searched refs:InterpreterWrapper (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
H A Dinterpreter_wrapper_pybind11.cc27 using tflite::interpreter_wrapper::InterpreterWrapper;
43 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromFile( in PYBIND11_MODULE()
58 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromFile( in PYBIND11_MODULE()
71 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromBuffer( in PYBIND11_MODULE()
86 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromBuffer( in PYBIND11_MODULE()
94 py::class_<InterpreterWrapper>(m, "InterpreterWrapper") in PYBIND11_MODULE()
97 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE()
105 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE()
110 [](const InterpreterWrapper& self) { in PYBIND11_MODULE()
114 [](InterpreterWrapper& self) { in PYBIND11_MODULE()
[all …]
H A Dinterpreter_wrapper.cc93 const InterpreterWrapper::Model* model, in CreateInterpreter()
198 InterpreterWrapper* InterpreterWrapper::CreateInterpreterWrapper( in CreateInterpreterWrapper()
199 std::unique_ptr<InterpreterWrapper::Model> model, int op_resolver_id, in CreateInterpreterWrapper()
242 InterpreterWrapper* wrapper = in CreateInterpreterWrapper()
243 new InterpreterWrapper(std::move(model), std::move(error_reporter), in CreateInterpreterWrapper()
248 InterpreterWrapper::InterpreterWrapper( in InterpreterWrapper() function in tflite::interpreter_wrapper::InterpreterWrapper
249 std::unique_ptr<InterpreterWrapper::Model> model, in InterpreterWrapper()
258 InterpreterWrapper::~InterpreterWrapper() {} in ~InterpreterWrapper()
263 PyObject* InterpreterWrapper::AllocateTensors(int subgraph_index) { in AllocateTensors()
274 PyObject* InterpreterWrapper::Invoke(int subgraph_index) { in Invoke()
[all …]
H A Dinterpreter_wrapper.h43 class InterpreterWrapper {
48 static InterpreterWrapper* CreateWrapperCPPFromFile(
52 static InterpreterWrapper* CreateWrapperCPPFromFile(
59 static InterpreterWrapper* CreateWrapperCPPFromBuffer(
63 static InterpreterWrapper* CreateWrapperCPPFromBuffer(
69 ~InterpreterWrapper();
116 static InterpreterWrapper* CreateInterpreterWrapper(
123 InterpreterWrapper(std::unique_ptr<Model> model,
130 InterpreterWrapper();
131 InterpreterWrapper(const InterpreterWrapper& rhs);
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/core/
H A Dtflite_engine.h55 using InterpreterWrapper = tflite::support::TfLiteInterpreterWrapper;
60 using InterpreterDeleter = InterpreterWrapper::InterpreterDeleter;
134 InterpreterWrapper* interpreter_wrapper() { return &interpreter_; } in interpreter_wrapper()
204 InterpreterWrapper interpreter_;
H A Dbase_task_api.h97 tflite::task::core::TfLiteEngine::InterpreterWrapper* interpreter_wrapper = in Infer()
117 tflite::task::core::TfLiteEngine::InterpreterWrapper* interpreter_wrapper = in InferWithFallback()
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter.h64 class InterpreterWrapper; // Class for friend declarations. variable
740 friend class tflite::interpreter_wrapper::InterpreterWrapper;