Home
last modified time | relevance | path

Searched refs:model_buffer (Results 1 – 25 of 72) sorted by relevance

123

/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_util_wrapper_pybind11.cc25 const std::vector<uint8_t>& model_buffer, in WrappedSetSignatureDefMap() argument
28 reinterpret_cast<const char*>(model_buffer.data()), model_buffer.size()); in WrappedSetSignatureDefMap()
50 const std::vector<uint8_t>& model_buffer) { in WrappedGetSignatureDefMap() argument
52 reinterpret_cast<const char*>(model_buffer.data()), model_buffer.size()); in WrappedGetSignatureDefMap()
71 py::bytes WrappedClearSignatureDefs(const std::vector<uint8_t>& model_buffer) { in WrappedClearSignatureDefs() argument
73 reinterpret_cast<const char*>(model_buffer.data()), model_buffer.size()); in WrappedClearSignatureDefs()
Dsignature_def_utils.py45 model_buffer = signature_def_util.SetSignatureDefMap(
47 return model_buffer
/external/tflite-support/tensorflow_lite_support/cc/test/task/text/
Dbert_nl_classifier_test.cc57 std::string model_buffer = LoadBinaryContent(test_model_path.c_str()); in TEST() local
59 BertNLClassifier::CreateFromBuffer(model_buffer.data(), model_buffer.size()); in TEST()
106 std::string model_buffer = LoadBinaryContent(test_model_path.c_str()); in TEST() local
108 BertNLClassifier::CreateFromBuffer(model_buffer.data(), model_buffer.size()); in TEST()
116 std::string model_buffer = LoadBinaryContent(test_model_path.c_str()); in TEST() local
118 BertNLClassifier::CreateFromBuffer(model_buffer.data(), model_buffer.size()); in TEST()
146 std::string model_buffer = LoadBinaryContent(test_model_path.c_str()); in TEST() local
155 BertNLClassifier::CreateFromBuffer(model_buffer.data(), model_buffer.size()); in TEST()
/external/libtextclassifier/native/annotator/
Dannotator_test-include.cc53 std::string model_buffer = ReadFile(GetTestModelPath()); in GetTestModelWithDatetimeRegEx() local
54 model_buffer = ModifyAnnotatorModel(model_buffer, [](ModelT* model) { in GetTestModelWithDatetimeRegEx()
57 return model_buffer; in GetTestModelWithDatetimeRegEx()
1336 std::string model_buffer = GetTestModelWithDatetimeRegEx(); in TEST_F() local
1338 Annotator::FromUnownedBuffer(model_buffer.data(), model_buffer.size(), in TEST_F()
1344 std::string model_buffer = ReadFile(GetTestModelPath()); in TEST_F() local
1345 model_buffer = ModifyAnnotatorModel(model_buffer, [](ModelT* model) { in TEST_F()
1349 Annotator::FromUnownedBuffer(model_buffer.data(), model_buffer.size(), in TEST_F()
1372 std::string model_buffer = ReadFile(GetTestModelPath()); in TEST_F() local
1373 model_buffer = ModifyAnnotatorModel(model_buffer, [](ModelT* model) { in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/common/
Ddelegate_compatibility_checker_base.cc34 tflite::FlatBufferModel* model_buffer, proto::CompatibilityResult* result) { in checkModelCompatibilityOffline() argument
35 auto model = tflite::GetModel(model_buffer); in checkModelCompatibilityOffline()
55 tflite::FlatBufferModel* model_buffer, proto::CompatibilityResult* result) { in checkModelCompatibilityOnline() argument
56 auto model = tflite::GetModel(model_buffer); in checkModelCompatibilityOnline()
Ddelegate_compatibility_checker_base.h46 tflite::FlatBufferModel* model_buffer,
54 tflite::FlatBufferModel* model_buffer,
/external/libtextclassifier/native/utils/grammar/semantics/
Dcomposer_test.cc80 const std::string model_buffer = PackFlatbuffer<RulesSet>(&model); in TEST_F() local
82 flatbuffers::GetRoot<RulesSet>(model_buffer.data())); in TEST_F()
156 const std::string model_buffer = PackFlatbuffer<RulesSet>(&model); in TEST_F() local
158 flatbuffers::GetRoot<RulesSet>(model_buffer.data())); in TEST_F()
/external/tflite-support/tensorflow_lite_support/cc/text/tokenizers/
Dsentencepiece_jni.cc37 JNIEnv* env, jobject obj, jobject model_buffer) { in Java_org_tensorflow_lite_support_text_tokenizers_SentencePieceTokenizer_nativeLoadResource() argument
38 auto model = GetMappedFileBuffer(env, model_buffer); in Java_org_tensorflow_lite_support_text_tokenizers_SentencePieceTokenizer_nativeLoadResource()
/external/libtextclassifier/native/utils/grammar/
Danalyzer_test.cc58 const std::string model_buffer = PackFlatbuffer<RulesSet>(&model); in TEST_F() local
61 flatbuffers::GetRoot<RulesSet>(model_buffer.data())); in TEST_F()
/external/tflite-support/tensorflow_lite_support/codegen/python/
Dcodegen.py70 model_buffer = get_model_buffer(FLAGS.model)
74 result = codegen.generate(model_buffer, FLAGS.package_name,
/external/tflite-support/tensorflow_lite_support/java/src/native/task/text/nlclassifier/bert_nlclassifier/
Dbert_nl_classifier_jni.cc48 JNIEnv* env, jclass thiz, jobject model_buffer) { in Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_initJniWithByteBuffer() argument
49 auto model = GetMappedFileBuffer(env, model_buffer); in Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_initJniWithByteBuffer()
/external/libtextclassifier/native/annotator/pod_ner/
Dpod-ner-impl_test.cc254 std::string model_buffer = in TEST_F() local
259 flatbuffers::GetRoot<PodNerModel>(model_buffer.data())), in TEST_F()
303 std::string model_buffer = in TEST_F() local
308 flatbuffers::GetRoot<PodNerModel>(model_buffer.data())), in TEST_F()
337 std::string model_buffer = in TEST_F() local
342 flatbuffers::GetRoot<PodNerModel>(model_buffer.data())), in TEST_F()
/external/tensorflow/tensorflow/lite/c/
Dc_api_test.cc250 std::vector<char> model_buffer(model_file.tellg()); in TEST() local
253 model_file.read(model_buffer.data(), model_buffer.size()); in TEST()
256 TfLiteModelCreate(model_buffer.data(), model_buffer.size()); in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc76 flatbuffers::Offset<Model> model_buffer = CreateModel( in Build() local
81 builder.Finish(model_buffer); in Build()
/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/detector/
Dobject_detector_jni.cc199 JNIEnv* env, jclass thiz, jobject model_buffer, jobject java_options) { in Java_org_tensorflow_lite_task_vision_detector_ObjectDetector_initJniWithByteBuffer() argument
203 static_cast<char*>(env->GetDirectBufferAddress(model_buffer)), in Java_org_tensorflow_lite_task_vision_detector_ObjectDetector_initJniWithByteBuffer()
204 static_cast<size_t>(env->GetDirectBufferCapacity(model_buffer))); in Java_org_tensorflow_lite_task_vision_detector_ObjectDetector_initJniWithByteBuffer()
/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/classifier/
Dimage_classifier_jni.cc192 JNIEnv* env, jclass thiz, jobject model_buffer, jobject java_options) { in Java_org_tensorflow_lite_task_vision_classifier_ImageClassifier_initJniWithByteBuffer() argument
200 static_cast<char*>(env->GetDirectBufferAddress(model_buffer)), in Java_org_tensorflow_lite_task_vision_classifier_ImageClassifier_initJniWithByteBuffer()
201 static_cast<size_t>(env->GetDirectBufferCapacity(model_buffer))); in Java_org_tensorflow_lite_task_vision_classifier_ImageClassifier_initJniWithByteBuffer()
/external/tflite-support/tensorflow_lite_support/java/src/native/task/vision/segmenter/
Dimage_segmenter_jni.cc206 JNIEnv* env, jclass thiz, jobject model_buffer, in Java_org_tensorflow_lite_task_vision_segmenter_ImageSegmenter_initJniWithByteBuffer() argument
211 static_cast<char*>(env->GetDirectBufferAddress(model_buffer)), in Java_org_tensorflow_lite_task_vision_segmenter_ImageSegmenter_initJniWithByteBuffer()
212 static_cast<size_t>(env->GetDirectBufferCapacity(model_buffer))); in Java_org_tensorflow_lite_task_vision_segmenter_ImageSegmenter_initJniWithByteBuffer()
/external/tflite-support/tensorflow_lite_support/java/src/native/task/text/nlclassifier/
Dnl_classifier_jni.cc94 jobject model_buffer) { in Java_org_tensorflow_lite_task_text_nlclassifier_NLClassifier_initJniWithByteBuffer() argument
95 auto model = GetMappedFileBuffer(env, model_buffer); in Java_org_tensorflow_lite_task_text_nlclassifier_NLClassifier_initJniWithByteBuffer()
/external/XNNPACK/bench/
Dconvert.cc472 flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_convert_f16_f32() local
479 builder.Finish(model_buffer); in tflite_convert_f16_f32()
571 flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_convert_f32_qs8() local
578 builder.Finish(model_buffer); in tflite_convert_f32_qs8()
672 flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_convert_f32_qu8() local
679 builder.Finish(model_buffer); in tflite_convert_f32_qu8()
775 flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_convert_qs8_f32() local
782 builder.Finish(model_buffer); in tflite_convert_qs8_f32()
878 flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_convert_qu8_f32() local
885 builder.Finish(model_buffer); in tflite_convert_qu8_f32()
Delu.cc208 const flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_elu_f32() local
215 builder.Finish(model_buffer); in tflite_elu_f32()
314 const flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_elu_qs8() local
321 builder.Finish(model_buffer); in tflite_elu_qs8()
Dsigmoid.cc347 const flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_sigmoid_f32() local
354 builder.Finish(model_buffer); in tflite_sigmoid_f32()
453 const flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_sigmoid_qs8() local
460 builder.Finish(model_buffer); in tflite_sigmoid_qs8()
559 const flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel(builder, in tflite_sigmoid_qu8() local
566 builder.Finish(model_buffer); in tflite_sigmoid_qu8()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dleaky_relu_tester.cc142 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
147 builder.Finish(model_buffer); in CreateTfLiteModel()
Dsoftmax_tester.cc144 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
149 builder.Finish(model_buffer); in CreateTfLiteModel()
Dreduce_tester.cc159 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
164 builder.Finish(model_buffer); in CreateTfLiteModel()
Dunary_elementwise_tester.cc169 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
174 builder.Finish(model_buffer); in CreateTfLiteModel()

123