TFlite在不使用FlexDelegate的情况下在C++推理中选择操作

#include <tensorflow/lite/interpreter.h>
#include <tensorflow/lite/interpreter_builder.h>
#include <tensorflow/lite/model.h>

// 加载模型
std::unique_ptr<tflite::FlatBufferModel> model = tflite::FlatBufferModel::BuildFromFile(model_path);

// 创建Interpreter
tflite::ops::builtin::BuiltinOpResolver resolver;
tflite::InterpreterBuilder builder(*model, resolver);
std::unique_ptr<tflite::Interpreter> interpreter;
builder(&interpreter);

// 设置操作选择策略为速度优先
tflite::InterpreterOptions options;
options.SetExecutionPlan(tflite::InterpreterOptions::kSpeed);

// 应用操作选择策略
interpreter->ModifyGraphWithDelegate(&options);

// 进行推理操作
interpreter->Invoke();