Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A Dcalibrator_test.cc201 auto flatbuffer_model = ReadModel("multi_add.bin"); in TEST() local
202 ASSERT_TRUE(flatbuffer_model); in TEST()
205 auto status = BuildLoggingInterpreter(*flatbuffer_model, in TEST()
209 auto readonly_model = flatbuffer_model->GetModel(); in TEST()
286 auto flatbuffer_model = ReadModel("lstm.bin"); in TEST() local
287 ASSERT_TRUE(flatbuffer_model); in TEST()
290 auto status = BuildLoggingInterpreter(*flatbuffer_model, in TEST()
295 auto readonly_model = flatbuffer_model->GetModel(); in TEST()
350 auto flatbuffer_model = ReadModel("unidirectional_sequence_lstm.bin"); in TEST() local
351 ASSERT_TRUE(flatbuffer_model); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/signature/
H A Dsignature_def_util_wrapper_pybind11.cc27 auto flatbuffer_model = tflite::FlatBufferModel::BuildFromBuffer( in WrappedSetSignatureDefMap() local
29 auto* model = flatbuffer_model->GetModel(); in WrappedSetSignatureDefMap()
51 auto flatbuffer_model = tflite::FlatBufferModel::BuildFromBuffer( in WrappedGetSignatureDefMap() local
53 auto* model = flatbuffer_model->GetModel(); in WrappedGetSignatureDefMap()
72 auto flatbuffer_model = tflite::FlatBufferModel::BuildFromBuffer( in WrappedClearSignatureDefs() local
74 auto* model = flatbuffer_model->GetModel(); in WrappedClearSignatureDefs()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/python/
H A Dtoco_python_api.cc81 std::unique_ptr<toco::Model> flatbuffer_model = in PopulateConversionLogHelper() local
85 PopulateConversionLog(*flatbuffer_model, &toco_log_after); in PopulateConversionLogHelper()
92 toco::LogDump(toco::kLogLevelModelChanged, "tflite_graph", *flatbuffer_model); in PopulateConversionLogHelper()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dtest_utils.py242 def load_model_from_flatbuffer(flatbuffer_model): argument
244 model = schema_fb.Model.GetRootAsModel(flatbuffer_model, 0)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc68 auto flatbuffer_model = local
70 flatbuffer_model->GetModel()->UnPackTo(model);