| /external/tensorflow/tensorflow/lite/core/api/ |
| D | op_resolver.cc | 25 auto builtin_code = opcode->builtin_code(); in GetRegistrationFromOpCode() local
28 if (builtin_code > BuiltinOperator_MAX || in GetRegistrationFromOpCode()
29 builtin_code < BuiltinOperator_MIN) { in GetRegistrationFromOpCode()
33 builtin_code); in GetRegistrationFromOpCode()
35 } else if (builtin_code != BuiltinOperator_CUSTOM) { in GetRegistrationFromOpCode()
36 *registration = op_resolver.FindOp(builtin_code, version); in GetRegistrationFromOpCode()
40 EnumNameBuiltinOperator(builtin_code), version); in GetRegistrationFromOpCode()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 224 op_idx, EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
234 op_idx, EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
239 op_idx, EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
246 EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
253 EnumNameBuiltinOperator(opcode->builtin_code())); in VerifySubGraphConsistency()
339 if (opcode->builtin_code() < BuiltinOperator_MIN || in VerifyOps()
340 opcode->builtin_code() > BuiltinOperator_MAX) { in VerifyOps()
342 opcode->builtin_code()); in VerifyOps()
346 if (opcode->builtin_code() == BuiltinOperator_CUSTOM) { in VerifyOps()
353 if (!resolver.FindOp(opcode->builtin_code(), opcode->version())) { in VerifyOps()
[all …]
|
| D | gen_op_registration.cc | 38 if (opcode->builtin_code() != ::tflite::BuiltinOperator_CUSTOM) { in ReadOpsFromModel()
40 tflite::EnumNameBuiltinOperator(opcode->builtin_code())); in ReadOpsFromModel()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_mutable_op_resolver.cc | 24 if ((registration.builtin_code == op) && in FindOp()
36 if ((registration.builtin_code == -1) && in FindOp()
57 new_registration->builtin_code = op; in AddBuiltin()
74 new_registration->builtin_code = -1; in AddCustom()
|
| D | micro_interpreter.cc | 43 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in OpNameFromRegistration()
46 return EnumNameBuiltinOperator(BuiltinOperator(registration->builtin_code)); in OpNameFromRegistration()
185 static_cast<BuiltinOperator>(registration->builtin_code); in Invoke()
207 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in Invoke()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 75 bool IsHybridOperator(const TfLiteContext* context, int builtin_code, in IsHybridOperator() argument
77 switch (builtin_code) { in IsHybridOperator()
528 static MappingFn Map(const TfLiteContext* context, int builtin_code, in Map() argument
531 switch (builtin_code) { in Map()
587 IsHybridOperator(context, builtin_code, node)) { in Map()
629 IsHybridOperator(context, builtin_code, node)) { in Map()
1170 int builtin_code, in AddDequantizeOperatorsWhereNeeded() argument
1183 switch (builtin_code) { in AddDequantizeOperatorsWhereNeeded()
1230 const bool hybrid_op = IsHybridOperator(context, reg->builtin_code, node); in AddOpsAndTensors()
1235 (reg->builtin_code == kTfLiteBuiltinLstm || in AddOpsAndTensors()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | logging_op_resolver_test.cc | 67 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CONV_2D); in TEST()
73 EXPECT_EQ(reg->builtin_code, BuiltinOperator_ADD); in TEST()
124 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CONV_2D); in TEST()
|
| D | calibrator.cc | 242 return tflite::EnumNamesBuiltinOperator()[opcode.builtin_code()]; in GetOpName()
292 op_info.builtin_op_code = operator_code->builtin_code(); in BuildLoggingInterpreter()
305 op_info.registration = op_resolver.FindOp(operator_code->builtin_code(), in BuildLoggingInterpreter()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 153 model->operator_codes[op->opcode_index]->builtin_code; in InsertQuantizableInputTensorsFromOperator()
198 if (model->operator_codes[i]->builtin_code == BuiltinOperator_DEQUANTIZE) { in GetOrInsertDequantizeOpCodeIndex()
204 model->operator_codes[op_code_idx]->builtin_code = BuiltinOperator_DEQUANTIZE; in GetOrInsertDequantizeOpCodeIndex()
236 const BuiltinOperator& op_code = model->operator_codes[i]->builtin_code; in UpdateInt8OperatorVersions()
297 model->operator_codes[consumer_op->opcode_index]->builtin_code; in QuantizeWeightsInternal()
|
| D | quantize_weights_test.cc | 101 *op_code = model->operator_codes()->Get(op_code_idx)->builtin_code(); in GetProducerOpCode()
176 ASSERT_EQ(output_model->operator_codes()->Get(op_code_idx)->builtin_code(), in TEST_F()
229 if (output_model->operator_codes()->Get(op_code_idx)->builtin_code() == in TEST_F()
281 output_model->operator_codes()->Get(op_code_idx)->builtin_code(); in TEST_F()
315 output_model->operator_codes()->Get(op_code_idx)->builtin_code(); in TEST_F()
|
| D | quantize_model_test.cc | 100 EXPECT_EQ(model_.operator_codes[i]->builtin_code, in TEST_F()
101 float_model_op->builtin_code()); in TEST_F()
|
| D | quantize_model.cc | 46 model->operator_codes[op->opcode_index]->builtin_code; in QuantizeModel()
|
| D | subgraph_quantizer_test.cc | 291 ASSERT_EQ(model.operator_codes[0].get()->builtin_code, in TEST()
348 ASSERT_EQ(model.operator_codes[0].get()->builtin_code, in TEST()
408 ASSERT_EQ(model.operator_codes[op->opcode_index].get()->builtin_code, in TEST()
463 ASSERT_EQ(model.operator_codes[op->opcode_index].get()->builtin_code, in TEST()
|
| /external/tensorflow/tensorflow/lite/ |
| D | mutable_op_resolver.cc | 38 new_registration.builtin_code = op; in AddBuiltin()
50 new_registration.builtin_code = BuiltinOperator_CUSTOM; in AddCustom()
|
| D | mutable_op_resolver_test.cc | 61 EXPECT_EQ(found_registration->builtin_code, BuiltinOperator_ADD); in TEST()
112 EXPECT_EQ(found_registration->builtin_code, BuiltinOperator_CUSTOM); in TEST()
|
| D | optional_debug_tools.cc | 109 reg.builtin_code); in PrintInterpreterState()
|
| D | model.cc | 274 static_cast<BuiltinOperator>(registration->builtin_code); in ParseNodes()
449 if ((registration->builtin_code == BuiltinOperator_CUSTOM) && in ApplyDelegates()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 96 reg.builtin_code = BuiltinOperator_MUL; in AddTfLiteMulOp()
170 reg.builtin_code = BuiltinOperator_CUSTOM; in AddTfOp()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | import.cc | 45 if (opcode->builtin_code() != ::tflite::BuiltinOperator_CUSTOM) { in LoadOperatorsTable()
47 EnumNameBuiltinOperator(opcode->builtin_code())); in LoadOperatorsTable()
|
| D | export_test.cc | 162 if (opcode->builtin_code() != ::tflite::BuiltinOperator_CUSTOM) { in ExportAndSummarizeOperators()
164 opcode->builtin_code())); in ExportAndSummarizeOperators()
481 (*operator_codes)[0]->builtin_code()); in TEST_F()
484 (*operator_codes)[1]->builtin_code()); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/examples/label_image/ |
| D | label_image.cc | 84 << registration.builtin_code << ", " in PrintProfilingInfo()
86 static_cast<BuiltinOperator>(registration.builtin_code)) in PrintProfilingInfo()
|
| /external/tensorflow/tensorflow/lite/java/src/test/native/ |
| D | interpreter_test_jni.cc | 48 .builtin_code = 0, in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
|
| /external/tensorflow/tensorflow/lite/experimental/writer/ |
| D | writer_lib.cc | 61 GetOpCodeForBuiltin(registration->builtin_code); in ExportOperators()
84 fbb, static_cast<enum BuiltinOperator>(registration.builtin_code), in ExportOperators()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer.cc | 71 int code = node_reg->second.builtin_code; in GetOperatorDetails()
|
| /external/tensorflow/tensorflow/lite/g3doc/performance/ |
| D | delegates.md | 63 switch (registration->builtin_code) {
92 kernel_registration.builtin_code = kTfLiteBuiltinDelegate;
|