Home
last modified time | relevance | path

Searched refs:FlatBufferModel (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/lite/
Dmodel.h63 class FlatBufferModel {
69 static std::unique_ptr<FlatBufferModel> BuildFromFile(
82 static std::unique_ptr<FlatBufferModel> VerifyAndBuildFromFile(
94 static std::unique_ptr<FlatBufferModel> BuildFromBuffer(
108 static std::unique_ptr<FlatBufferModel> VerifyAndBuildFromBuffer(
118 static std::unique_ptr<FlatBufferModel> BuildFromModel(
123 ~FlatBufferModel();
126 FlatBufferModel(const FlatBufferModel&) = delete;
127 FlatBufferModel& operator=(const FlatBufferModel&) = delete;
145 FlatBufferModel(std::unique_ptr<Allocation> allocation,
[all …]
Dmodel.cc82 std::unique_ptr<FlatBufferModel> FlatBufferModel::BuildFromFile( in BuildFromFile()
86 std::unique_ptr<FlatBufferModel> model; in BuildFromFile()
89 model.reset(new FlatBufferModel(std::move(allocation), error_reporter)); in BuildFromFile()
94 std::unique_ptr<FlatBufferModel> FlatBufferModel::VerifyAndBuildFromFile( in VerifyAndBuildFromFile()
99 std::unique_ptr<FlatBufferModel> model; in VerifyAndBuildFromFile()
116 model.reset(new FlatBufferModel(std::move(allocation), error_reporter)); in VerifyAndBuildFromFile()
122 std::unique_ptr<FlatBufferModel> FlatBufferModel::BuildFromBuffer( in BuildFromBuffer()
127 std::unique_ptr<FlatBufferModel> model; in BuildFromBuffer()
130 model.reset(new FlatBufferModel(std::move(allocation), error_reporter)); in BuildFromBuffer()
135 std::unique_ptr<FlatBufferModel> FlatBufferModel::VerifyAndBuildFromBuffer( in VerifyAndBuildFromBuffer()
[all …]
Dmodel_test.cc72 ASSERT_TRUE(!FlatBufferModel::BuildFromFile("/tmp/tflite_model_1234")); in TEST()
77 auto model = FlatBufferModel::BuildFromFile( in TEST()
91 auto m = FlatBufferModel::BuildFromFile( in TEST()
99 auto m = FlatBufferModel::BuildFromFile( in TEST()
109 auto model = FlatBufferModel::BuildFromFile( in TEST()
121 auto model = FlatBufferModel::BuildFromFile( in TEST()
201 auto model = FlatBufferModel::BuildFromFile( in TEST()
222 ASSERT_FALSE(FlatBufferModel::BuildFromFile( in TEST()
250 ASSERT_TRUE(FlatBufferModel::VerifyAndBuildFromFile( in TEST()
257 ASSERT_FALSE(FlatBufferModel::VerifyAndBuildFromFile( in TEST()
[all …]
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.h37 class FlatBufferModel; variable
80 std::unique_ptr<tflite::FlatBufferModel> model,
85 std::unique_ptr<tflite::FlatBufferModel> model,
98 const std::unique_ptr<tflite::FlatBufferModel> model_;
Dinterpreter_wrapper.cc56 const tflite::FlatBufferModel* model, in CreateInterpreter()
87 std::unique_ptr<tflite::FlatBufferModel> model, in CreateInterpreterWrapper()
109 std::unique_ptr<tflite::FlatBufferModel> model, in InterpreterWrapper()
411 std::unique_ptr<tflite::FlatBufferModel> model = in CreateWrapperCPPFromFile()
412 tflite::FlatBufferModel::BuildFromFile(model_path, error_reporter.get()); in CreateWrapperCPPFromFile()
426 std::unique_ptr<tflite::FlatBufferModel> model = in CreateWrapperCPPFromBuffer()
427 tflite::FlatBufferModel::BuildFromBuffer(buf, length, in CreateWrapperCPPFromBuffer()
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_test.cc33 std::unique_ptr<tflite::FlatBufferModel> model = in main()
34 tflite::FlatBufferModel::BuildFromFile(argv[1]); in main()
44 model = tflite::FlatBufferModel::BuildFromBuffer( in main()
Dwriter.cc32 std::unique_ptr<tflite::FlatBufferModel> model = in main()
33 tflite::FlatBufferModel::BuildFromFile(argv[1]); in main()
Dwriter_lib_test.cc50 std::unique_ptr<FlatBufferModel> model = in TEST()
51 FlatBufferModel::BuildFromFile("/tmp/test.tflite"); in TEST()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.h37 class FlatBufferModel; variable
72 std::unique_ptr<tflite::FlatBufferModel> model,
83 std::unique_ptr<tflite::FlatBufferModel> model_;
Dcalibration_wrapper.cc64 std::unique_ptr<tflite::FlatBufferModel> model, in CalibrationWrapper()
188 std::unique_ptr<tflite::FlatBufferModel> model = in CreateWrapperCPPFromBuffer()
189 tflite::FlatBufferModel::BuildFromBuffer(buf, length, in CreateWrapperCPPFromBuffer()
/external/libtextclassifier/utils/
Dtflite-model-executor.h35 std::unique_ptr<const tflite::FlatBufferModel> TfLiteModelFromModelSpec(
37 std::unique_ptr<const tflite::FlatBufferModel> TfLiteModelFromBuffer(
134 std::unique_ptr<const tflite::FlatBufferModel> model);
136 std::unique_ptr<const tflite::FlatBufferModel> model_;
Dtflite-model-executor.cc178 std::unique_ptr<const tflite::FlatBufferModel> TfLiteModelFromModelSpec( in TfLiteModelFromModelSpec()
180 std::unique_ptr<const tflite::FlatBufferModel> model( in TfLiteModelFromModelSpec()
181 tflite::FlatBufferModel::BuildFromModel(model_spec)); in TfLiteModelFromModelSpec()
189 std::unique_ptr<const tflite::FlatBufferModel> TfLiteModelFromBuffer( in TfLiteModelFromBuffer()
202 std::unique_ptr<const tflite::FlatBufferModel> model) in TfLiteModelExecutor()
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api.cc53 auto model = tflite::FlatBufferModel::BuildFromBuffer( in TFL_NewModel()
55 std::shared_ptr<const tflite::FlatBufferModel> shared_model(model.release()); in TFL_NewModel()
60 auto model = tflite::FlatBufferModel::BuildFromFile(model_path); in TFL_NewModelFromFile()
61 std::shared_ptr<const tflite::FlatBufferModel> shared_model(model.release()); in TFL_NewModelFromFile()
Dc_api_internal.h32 std::shared_ptr<const tflite::FlatBufferModel> impl;
51 std::shared_ptr<const tflite::FlatBufferModel> model;
/external/tensorflow/tensorflow/lite/examples/minimal/
Dminimal.cc47 std::unique_ptr<tflite::FlatBufferModel> model = in main()
48 tflite::FlatBufferModel::BuildFromFile(filename); in main()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dinference.md80 The `FlatBufferModel` class encapsulates a model and can be built in a couple of
84 class FlatBufferModel {
86  static std::unique_ptr<FlatBufferModel> BuildFromFile(
93  static std::unique_ptr<FlatBufferModel> BuildFromBuffer(
101 tflite::FlatBufferModel model(path_to_model);
105 automatically try to use shared memory to store the FlatBufferModel.
138 * Build an `Interpreter` based on an existing `FlatBufferModel`
154 model into a `FlatBufferModel` object which then can be executed by an
155 `Interpreter`. The `FlatBufferModel` needs to remain valid for the whole
156 lifetime of the `Interpreter`, and a single `FlatBufferModel` can be
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model_test.cc39 std::unique_ptr<FlatBufferModel> ReadTestModel() { in ReadTestModel()
42 return FlatBufferModel::BuildFromFile(model_path.c_str()); in ReadTestModel()
58 std::unique_ptr<FlatBufferModel> input_model_;
Dquantize_weights_test.cc39 std::unique_ptr<FlatBufferModel> ReadTestModel() { in ReadTestModel()
42 return FlatBufferModel::BuildFromFile(model_path.c_str()); in ReadTestModel()
45 std::unique_ptr<FlatBufferModel> ReadSharedWeightsTestModel() { in ReadSharedWeightsTestModel()
48 return FlatBufferModel::BuildFromFile(model_path.c_str()); in ReadSharedWeightsTestModel()
70 std::unique_ptr<FlatBufferModel> input_model_;
Dsubgraph_quantizer_test.cc35 std::unique_ptr<FlatBufferModel> ReadModel(const char* model) { in ReadModel()
37 return FlatBufferModel::BuildFromFile(model_path.c_str()); in ReadModel()
40 std::unique_ptr<FlatBufferModel> ReadConvModel1() { in ReadConvModel1()
44 std::unique_ptr<FlatBufferModel> ReadConvModel2() { in ReadConvModel2()
48 std::unique_ptr<FlatBufferModel> ReadSoftmaxModel() { in ReadSoftmaxModel()
52 std::unique_ptr<FlatBufferModel> ReadAvgPoolModel() { in ReadAvgPoolModel()
56 std::unique_ptr<FlatBufferModel> ReadMultiInputAddWithReshapeModel() { in ReadMultiInputAddWithReshapeModel()
Dquantization_utils_test.cc34 std::unique_ptr<FlatBufferModel> ReadModel(const char* model) { in ReadModel()
36 return FlatBufferModel::BuildFromFile(model_path.c_str()); in ReadModel()
39 std::unique_ptr<FlatBufferModel> ReadConvModel() { in ReadConvModel()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dnode_info_delegate_test.cc36 std::unique_ptr<FlatBufferModel> ReadModel(const char* model) { in ReadModel()
38 return FlatBufferModel::BuildFromFile(model_path.c_str()); in ReadModel()
41 std::unique_ptr<FlatBufferModel> ReadModel() { in ReadModel()
/external/tensorflow/tensorflow/lite/tools/accuracy/
Dutils.cc89 std::unique_ptr<tflite::FlatBufferModel> model; in GetTFliteModelInfo()
91 model = tflite::FlatBufferModel::BuildFromFile(model_file_path.data()); in GetTFliteModelInfo()
/external/tensorflow/tensorflow/lite/models/smartreply/
Dpredictor_test.cc63 model_ = tflite::FlatBufferModel::BuildFromFile(GetModelFilePath().c_str()); in SetUp()
67 std::unique_ptr<::tflite::FlatBufferModel> model_;
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc28 tflite::FlatBufferModel* convertLongToModel(JNIEnv* env, jlong handle) { in convertLongToModel()
34 return reinterpret_cast<tflite::FlatBufferModel*>(handle); in convertLongToModel()
298 auto model = tflite::FlatBufferModel::VerifyAndBuildFromFile( in Java_org_tensorflow_lite_NativeInterpreterWrapper_createModel()
327 auto model = tflite::FlatBufferModel::BuildFromBuffer( in Java_org_tensorflow_lite_NativeInterpreterWrapper_createModelWithBuffer()
342 tflite::FlatBufferModel* model = convertLongToModel(env, model_handle); in Java_org_tensorflow_lite_NativeInterpreterWrapper_createInterpreter()
/external/tensorflow/tensorflow/lite/models/smartreply/demo/app/src/main/
Dsmartreply_jni.cc53 std::unique_ptr<::tflite::FlatBufferModel> model;
65 storage->model = tflite::FlatBufferModel::BuildFromBuffer( in Java_com_example_android_smartreply_SmartReplyClient_loadJNI()

12