tflite::
Summary
Constructors and Destructors
InterpreterBuilder
(const FlatBufferModel & model, const OpResolver
& op_resolver, const InterpreterOptions *options_experimental)
InterpreterBuilder
(const ::tflite::Model *model, const OpResolver
& op_resolver, ErrorReporter
*error_reporter, const InterpreterOptions *options_experimental, const Allocation
*allocation)
InterpreterBuilder
(const InterpreterBuilder &)
Public functions
AddDelegate
(TfLiteDelegate *delegate)
void
AddDelegate
(TfLiteOpaqueDelegateStruct *opaque_delegate)
void
SetNumThreads
(int num_threads)
TfLiteStatus
SetTelemetryProfiler
(std::unique_ptr< telemetry::TelemetryProfiler > profiler)
void
operator()
(std::unique_ptr< Interpreter
> *interpreter)
TfLiteStatus
*interpreter
. operator()
(std::unique_ptr< Interpreter
> *interpreter, int num_threads)
TfLiteStatus
operator=
(const InterpreterBuilder
&)=delete
Public functions
AddDelegate
void AddDelegate( TfLiteDelegate *delegate )
Any delegates added with AddDelegate will be applied to the Interpreter generated by operator(), in the order that they were added.
(The delegate parameter passed to AddDelegate should be non-null, otherwise an error will be reported, and the call to AddDelegate will have no other effect.) The lifetime of the delegate must be at least as long as the lifetime of any Interpreter generated by this InterpreterBuilder.
AddDelegate
void AddDelegate( TfLiteOpaqueDelegateStruct *opaque_delegate )
InterpreterBuilder
InterpreterBuilder ( const FlatBufferModel & model , const OpResolver & op_resolver , const InterpreterOptions * options_experimental )
For this constructor, the ErrorReporter will be extracted from the FlatBufferModel.
options
object is copied during construction. So caller can release it after calling the constructor.
InterpreterBuilder
InterpreterBuilder ( const :: tflite :: Model * model , const OpResolver & op_resolver , ErrorReporter * error_reporter , const InterpreterOptions * options_experimental , const Allocation * allocation )
Builds an interpreter given only the raw flatbuffer Model object (instead of a FlatBufferModel).
Mostly used for testing. If error_reporter
is null, then DefaultErrorReporter() is used. options
object is copied during construction. So caller can release it after calling the constructor.
InterpreterBuilder
InterpreterBuilder ( const InterpreterBuilder & ) = delete
SetNumThreads
TfLiteStatus SetNumThreads( int num_threads )
Sets the number of CPU threads to use for the interpreter.
Returns kTfLiteOk on success, kTfLiteError on error.
SetTelemetryProfiler
void SetTelemetryProfiler( std::unique_ptr< telemetry::TelemetryProfiler > profiler )
operator()
TfLiteStatus operator()( std::unique_ptr< Interpreter > *interpreter )
Builds an interpreter and stores it in *interpreter
.
On success, returns kTfLiteOk and sets *interpreter
to a valid Interpreter. On failure, returns an error status and sets *interpreter
to nullptr.
operator()
TfLiteStatus operator()( std::unique_ptr< Interpreter > *interpreter, int num_threads )
Same as above, but also sets the number of CPU threads to use (overriding any previous call to SetNumThreads).
Deprecated: use the SetNumThreads method instead.
operator=
InterpreterBuilder & operator = ( const InterpreterBuilder & ) = delete
~InterpreterBuilder
~InterpreterBuilder()
