Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper_pybind11.cc24 using tflite::interpreter_wrapper::InterpreterWrapper;
39 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromFile( in PYBIND11_MODULE()
52 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromFile( in PYBIND11_MODULE()
63 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromBuffer( in PYBIND11_MODULE()
76 auto* wrapper = ::InterpreterWrapper::CreateWrapperCPPFromBuffer( in PYBIND11_MODULE()
83 py::class_<InterpreterWrapper>(m, "InterpreterWrapper") in PYBIND11_MODULE()
85 [](InterpreterWrapper& self) { in PYBIND11_MODULE()
89 [](InterpreterWrapper& self) { in PYBIND11_MODULE()
93 [](const InterpreterWrapper& self) { in PYBIND11_MODULE()
97 [](InterpreterWrapper& self) { in PYBIND11_MODULE()
[all …]
Dinterpreter_wrapper.cc70 const InterpreterWrapper::Model* model, in CreateInterpreter()
168 InterpreterWrapper* InterpreterWrapper::CreateInterpreterWrapper( in CreateInterpreterWrapper()
169 std::unique_ptr<InterpreterWrapper::Model> model, in CreateInterpreterWrapper()
193 InterpreterWrapper* wrapper = in CreateInterpreterWrapper()
194 new InterpreterWrapper(std::move(model), std::move(error_reporter), in CreateInterpreterWrapper()
199 InterpreterWrapper::InterpreterWrapper( in InterpreterWrapper() function in tflite::interpreter_wrapper::InterpreterWrapper
200 std::unique_ptr<InterpreterWrapper::Model> model, in InterpreterWrapper()
209 InterpreterWrapper::~InterpreterWrapper() {} in ~InterpreterWrapper()
211 PyObject* InterpreterWrapper::AllocateTensors() { in AllocateTensors()
217 PyObject* InterpreterWrapper::Invoke() { in Invoke()
[all …]
Dinterpreter_wrapper.h48 class InterpreterWrapper {
53 static InterpreterWrapper* CreateWrapperCPPFromFile(
56 static InterpreterWrapper* CreateWrapperCPPFromFile(
63 static InterpreterWrapper* CreateWrapperCPPFromBuffer(
66 static InterpreterWrapper* CreateWrapperCPPFromBuffer(
71 ~InterpreterWrapper();
121 static InterpreterWrapper* CreateInterpreterWrapper(
128 InterpreterWrapper(
136 InterpreterWrapper();
137 InterpreterWrapper(const InterpreterWrapper& rhs);
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtflite_engine.h57 using InterpreterWrapper = tflite::support::TfLiteInterpreterWrapper;
62 using InterpreterDeleter = InterpreterWrapper::InterpreterDeleter;
136 InterpreterWrapper* interpreter_wrapper() { return &interpreter_; } in interpreter_wrapper()
222 InterpreterWrapper interpreter_;
Dbase_task_api.h97 tflite::task::core::TfLiteEngine::InterpreterWrapper* interpreter_wrapper = in Infer()
117 tflite::task::core::TfLiteEngine::InterpreterWrapper* interpreter_wrapper = in InferWithFallback()