Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/micro/benchmarks/
Dkeyword_benchmark.cc54 KeywordOpResolver* op_resolver = new (op_resolver_buffer) KeywordOpResolver(); in CreateBenchmarkRunner() local
55 op_resolver->AddFullyConnected(tflite::Register_FULLY_CONNECTED_INT8()); in CreateBenchmarkRunner()
56 op_resolver->AddQuantize(); in CreateBenchmarkRunner()
57 op_resolver->AddSoftmax(); in CreateBenchmarkRunner()
58 op_resolver->AddSvdf(); in CreateBenchmarkRunner()
61 KeywordBenchmarkRunner(g_keyword_scrambled_model_data, op_resolver, in CreateBenchmarkRunner()
Dmicro_benchmark.h35 const tflite::MicroOpResolver* op_resolver, in MicroBenchmarkRunner() argument
38 : interpreter_(GetModel(model), *op_resolver, tensor_arena, in MicroBenchmarkRunner()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter_test.cc63 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
70 tflite::MicroInterpreter interpreter(model, op_resolver, allocator_buffer, in TF_LITE_MICRO_TEST()
113 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
125 tflite::MicroInterpreter interpreter0(model0, op_resolver, allocator, in TF_LITE_MICRO_TEST()
146 tflite::MicroInterpreter interpreter1(model1, op_resolver, allocator, in TF_LITE_MICRO_TEST()
158 tflite::MicroInterpreter interpreter2(model2, op_resolver, allocator, in TF_LITE_MICRO_TEST()
188 tflite::MicroInterpreter interpreter3(model3, op_resolver, allocator, in TF_LITE_MICRO_TEST()
209 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
221 tflite::MicroInterpreter interpreter(model, op_resolver, allocator, in TF_LITE_MICRO_TEST()
258 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
[all …]
Drecording_micro_interpreter.h38 const MicroOpResolver& op_resolver, in RecordingMicroInterpreter() argument
41 : MicroInterpreter(model, op_resolver, in RecordingMicroInterpreter()
49 const MicroOpResolver& op_resolver, in RecordingMicroInterpreter() argument
52 : MicroInterpreter(model, op_resolver, allocator, error_reporter), in RecordingMicroInterpreter()
Dmicro_allocator_test.cc240 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
249 allocator->StartModelAllocation(model, op_resolver, in TF_LITE_MICRO_TEST()
253 allocator->StartModelAllocation(model, op_resolver, in TF_LITE_MICRO_TEST()
261 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
277 allocator->StartModelAllocation(model, op_resolver, in TF_LITE_MICRO_TEST()
281 allocator->StartModelAllocation(model, op_resolver, in TF_LITE_MICRO_TEST()
289 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
298 allocator->StartModelAllocation(model, op_resolver, in TF_LITE_MICRO_TEST()
328 tflite::AllOpsResolver op_resolver = tflite::testing::GetOpResolver(); in TF_LITE_MICRO_TEST() local
345 allocator->StartModelAllocation(model1, op_resolver, in TF_LITE_MICRO_TEST()
[all …]
Dmicro_interpreter.cc108 const MicroOpResolver& op_resolver, in MicroInterpreter() argument
114 op_resolver_(op_resolver), in MicroInterpreter()
128 const MicroOpResolver& op_resolver, in MicroInterpreter() argument
133 op_resolver_(op_resolver), in MicroInterpreter()
Dmicro_allocator.h125 const Model* model, const MicroOpResolver& op_resolver,
202 const Model* model, const MicroOpResolver& op_resolver,
Dmicro_interpreter.h86 MicroInterpreter(const Model* model, const MicroOpResolver& op_resolver,
96 MicroInterpreter(const Model* model, const MicroOpResolver& op_resolver,
Drecording_micro_allocator.cc157 const Model* model, const MicroOpResolver& op_resolver, in PrepareNodeAndRegistrationDataFromFlatbuffer() argument
163 model, op_resolver, node_and_registrations); in PrepareNodeAndRegistrationDataFromFlatbuffer()
/external/tensorflow/tensorflow/lite/core/api/
DBUILD19 "op_resolver.h",
28 ":op_resolver",
39 # We define separate targets for "op_resolver" and "error_reporter",
46 name = "op_resolver",
47 srcs = ["op_resolver.cc"],
48 hdrs = ["op_resolver.h"],
Dop_resolver.cc26 const OperatorCode* opcode, const OpResolver& op_resolver, in GetRegistrationFromOpCode() argument
42 *registration = op_resolver.FindOp(builtin_code, version); in GetRegistrationFromOpCode()
59 *registration = op_resolver.FindOp(name, version); in GetRegistrationFromOpCode()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.h58 const FlatBufferModel& model, const OpResolver& op_resolver,
65 const OpResolver& op_resolver, std::unique_ptr<Interpreter>* interpreter,
Dcalibrator.cc321 const FlatBufferModel& model, const OpResolver& op_resolver, in BuildLoggingInterpreter() argument
325 op_resolver, interpreter, calibration_reader); in BuildLoggingInterpreter()
330 const OpResolver& op_resolver, std::unique_ptr<Interpreter>* interpreter, in BuildLoggingInterpreter() argument
380 op_resolver.FindOp(op_info.name.c_str(), operator_code->version()); in BuildLoggingInterpreter()
384 op_info.registration = op_resolver.FindOp(GetBuiltinCode(operator_code), in BuildLoggingInterpreter()
395 builtin_op_and_versions, custom_op_and_versions, op_resolver, LoggingEval, in BuildLoggingInterpreter()
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtflite_engine.cc217 OpResolver* op_resolver = reinterpret_cast<OpResolver*>(user_data); in FindBuiltinOp() local
219 return op_resolver->FindOp(op, version); in FindBuiltinOp()
224 OpResolver* op_resolver = reinterpret_cast<OpResolver*>(user_data); in FindCustomOp() local
225 return op_resolver->FindOp(custom_op, version); in FindCustomOp()
DBUILD16 "@org_tensorflow//tensorflow/lite/core/api:op_resolver",
72 "@org_tensorflow//tensorflow/lite/core/api:op_resolver",
113 "@org_tensorflow//tensorflow/lite/core/api:op_resolver",
Dtflite_engine.h197 explicit Verifier(const tflite::OpResolver* op_resolver) in Verifier() argument
198 : op_resolver_(op_resolver) {} in Verifier()
/external/tensorflow/tensorflow/lite/
Dinterpreter_builder.h61 const OpResolver& op_resolver);
66 const OpResolver& op_resolver,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/
Dtflite_model_reader.cc77 const tflite::OpResolver& op_resolver, in BuildFromFlatBuffer() argument
80 tflite::InterpreterBuilder interpreter_builder(flatbuffer, op_resolver); in BuildFromFlatBuffer()
Dinterpreter_utils.cc38 const OpResolver& op_resolver, const std::vector<TensorFloat32>& inputs, in InterpreterInvokeWithOpResolver() argument
41 if (InterpreterBuilder(model, op_resolver)(&interpreter) != kTfLiteOk) { in InterpreterInvokeWithOpResolver()
Dtflite_model_reader.h29 const tflite::OpResolver& op_resolver,
Dinterpreter_utils.h36 const OpResolver& op_resolver, const std::vector<TensorFloat32>& inputs,
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
Dperformance_profiling.cc36 ops::builtin::BuiltinOpResolver op_resolver; in RunModelSample() local
37 RETURN_IF_ERROR(BuildFromFlatBuffer(*flatbuffer, op_resolver, &graph_cl)); in RunModelSample()
Dinternal_api_samples.cc101 ops::builtin::BuiltinOpResolver op_resolver; in RunModelSampleWithInternalAPI() local
102 InterpreterBuilder tfl_builder(*flatbuffer, op_resolver); in RunModelSampleWithInternalAPI()
138 RETURN_IF_ERROR(BuildFromFlatBuffer(*flatbuffer, op_resolver, &graph_cl)); in RunModelSampleWithInternalAPI()
225 ops::builtin::BuiltinOpResolver op_resolver; in RunModelSampleWithInternalAPISerializedKernels() local
226 InterpreterBuilder tfl_builder(*flatbuffer, op_resolver); in RunModelSampleWithInternalAPISerializedKernels()
262 RETURN_IF_ERROR(BuildFromFlatBuffer(*flatbuffer, op_resolver, &graph_cl)); in RunModelSampleWithInternalAPISerializedKernels()
Ddelegate_testing.cc99 tflite::ops::builtin::BuiltinOpResolver op_resolver; in main() local
100 tflite::InterpreterBuilder builder(*model, op_resolver); in main()
/external/tensorflow/tensorflow/lite/c/
Dc_api.cc256 tflite::OpResolver* op_resolver = mutable_resolver; in InterpreterCreateWithOpResolver() local
268 op_resolver = &callback_op_resolver; in InterpreterCreateWithOpResolver()
274 tflite::InterpreterBuilder builder(model->impl->GetModel(), *op_resolver, in InterpreterCreateWithOpResolver()

12