Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_util_wrapper_pybind11.cc23 const std::vector<uint8_t>& model_buffer, in WrappedSetSignatureDefMap() argument
26 reinterpret_cast<const char*>(model_buffer.data()), model_buffer.size()); in WrappedSetSignatureDefMap()
48 const std::vector<uint8_t>& model_buffer) { in WrappedGetSignatureDefMap() argument
50 reinterpret_cast<const char*>(model_buffer.data()), model_buffer.size()); in WrappedGetSignatureDefMap()
69 py::bytes WrappedClearSignatureDefs(const std::vector<uint8_t>& model_buffer) { in WrappedClearSignatureDefs() argument
71 reinterpret_cast<const char*>(model_buffer.data()), model_buffer.size()); in WrappedClearSignatureDefs()
Dsignature_def_utils.py46 model_buffer = signature_def_util.SetSignatureDefMap(
48 return model_buffer
/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()
1364 std::string model_buffer = GetTestModelWithDatetimeRegEx(); in TEST_F() local
1366 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()
1377 Annotator::FromUnownedBuffer(model_buffer.data(), model_buffer.size(), in TEST_F()
1400 std::string model_buffer = ReadFile(GetTestModelPath()); in TEST_F() local
1401 model_buffer = ModifyAnnotatorModel(model_buffer, [](ModelT* model) { in TEST_F()
[all …]
/external/tflite-support/tensorflow_lite_support/cc/text/tokenizers/
Dtokenizer_utils.cc80 ASSIGN_OR_RETURN(absl::string_view model_buffer, in CreateTokenizerFromProcessUnit()
84 return absl::make_unique<SentencePieceTokenizer>(model_buffer.data(), in CreateTokenizerFromProcessUnit()
85 model_buffer.size()); in CreateTokenizerFromProcessUnit()
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/tensorflow/tensorflow/lite/c/
Dc_api_test.cc246 std::vector<char> model_buffer(model_file.tellg()); in TEST() local
249 model_file.read(model_buffer.data(), model_buffer.size()); in TEST()
252 TfLiteModelCreate(model_buffer.data(), model_buffer.size()); in TEST()
/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/java/src/native/task/text/nlclassifier/bert_nlclassifier/
Dbert_nl_classifier_jni.cc39 JNIEnv* env, jclass thiz, jobject model_buffer) { in Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_initJniWithByteBuffer() argument
40 auto model = GetMappedFileBuffer(env, model_buffer); in Java_org_tensorflow_lite_task_text_nlclassifier_BertNLClassifier_initJniWithByteBuffer()
/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/libtextclassifier/native/annotator/pod_ner/
Dpod-ner-impl_test.cc230 std::string model_buffer = in TEST_F() local
235 flatbuffers::GetRoot<PodNerModel>(model_buffer.data())), in TEST_F()
279 std::string model_buffer = in TEST_F() local
284 flatbuffers::GetRoot<PodNerModel>(model_buffer.data())), in TEST_F()
313 std::string model_buffer = in TEST_F() local
318 flatbuffers::GetRoot<PodNerModel>(model_buffer.data())), in TEST_F()
/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/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/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/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/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/tensorflow/tensorflow/lite/delegates/xnnpack/
Dsoftmax_tester.cc144 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
149 builder.Finish(model_buffer); in CreateTfLiteModel()
Dleaky_relu_tester.cc142 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
147 builder.Finish(model_buffer); in CreateTfLiteModel()
Dreshape_tester.cc169 const flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
174 builder.Finish(model_buffer); in CreateTfLiteModel()
Ddepth_to_space_tester.cc158 const 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()
Dreduce_tester.cc159 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
164 builder.Finish(model_buffer); in CreateTfLiteModel()
Dresize_bilinear_tester.cc176 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
181 builder.Finish(model_buffer); in CreateTfLiteModel()
Dpad_tester.cc175 flatbuffers::Offset<Model> model_buffer = CreateModel( in CreateTfLiteModel() local
180 builder.Finish(model_buffer); in CreateTfLiteModel()
Dpool_2d_tester.cc189 flatbuffers::Offset<tflite::Model> model_buffer = tflite::CreateModel( in CreateTfLiteModel() local
194 builder.Finish(model_buffer); in CreateTfLiteModel()

12