Home
last modified time | relevance | path

Searched refs:GetBuiltinCode (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model_test.cc195 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[0].get()), in TEST_P()
209 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[i].get()), in TEST_P()
210 GetBuiltinCode(float_model_op)); in TEST_P()
211 if (GetBuiltinCode(model_.operator_codes[i].get()) == in TEST_P()
276 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[quant_idx].get()), in TEST_P()
278 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[dequant_idx].get()), in TEST_P()
342 GetBuiltinCode(model_.operator_codes[quant_op_uint8_int8_idx].get()), in TEST_P()
345 GetBuiltinCode(model_.operator_codes[quant_op_int8_uint8_idx].get()), in TEST_P()
451 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[requant->opcode_index].get()), in TEST_P()
453 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[concat->opcode_index].get()), in TEST_P()
[all …]
Dquantize_weights_test.cc125 *op_code = GetBuiltinCode(model->operator_codes()->Get(op_code_idx)); in GetProducerOpCode()
200 ASSERT_EQ(GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)), in TEST_F()
259 if (GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)) == in TEST_F()
318 if (GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)) == in TEST_F()
370 GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)); in TEST_F()
404 GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)); in TEST_F()
444 GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)); in TEST_F()
484 GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)); in TEST_F()
530 GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)); in TEST_F()
565 ASSERT_EQ(GetBuiltinCode(output_model->operator_codes()->Get(op_code_idx)), in TEST_F()
Dquantize_weights.cc83 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in GetWeightInputIndices()
137 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in IsHybridEvaluationOp()
202 auto builtin_code = GetBuiltinCode(op_code); in InsertQuantizableInputTensorsFromOperator()
298 if (GetBuiltinCode(model->operator_codes[i].get()) == in GetOrInsertDequantizeOpCodeIndex()
344 GetBuiltinCode(model->operator_codes[i].get()); in UpdateInt8OperatorVersions()
377 GetBuiltinCode(model->operator_codes[consumer_op->opcode_index].get()); in IsQuantizationPassThroughOps()
Dquantization_wrapper_utils_test.cc74 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
101 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
Dquantization_wrapper_utils_custom_test.cc73 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
108 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
Dmodify_model_interface.cc69 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in GetInputTensors()
144 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in GetOutputTensors()
220 if (GetBuiltinCode(model->operator_codes[i].get()) == in SetOutputTypeToUINT8()
Dquantize_model.cc63 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in GetOperatorProperty()
231 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in InputQuantizeRequired()
605 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in QuantizeOpInput()
800 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in QuantizeOpOutput()
896 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in QuantizeIntemediateTensors()
1029 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in QuantizeWeightsInputOutput()
1094 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in QuantizeBiases()
Dquantization_utils_test.cc476 GetBuiltinCode(model.operator_codes.at(conv_op->opcode_index).get()), in TEST_F()
526 GetBuiltinCode(model.operator_codes.at(conv_op->opcode_index).get()), in TEST_F()
561 GetBuiltinCode(model.operator_codes.at(conv_op->opcode_index).get()), in TEST_F()
630 GetBuiltinCode(model.operator_codes.at(conv_op->opcode_index).get()), in TEST_F()
Dmodel_utils.cc40 if (GetBuiltinCode(model->operator_codes[i].get()) == op_code) { in GetOrInsertOpCodeIndex()
Dmodify_model_interface_test.cc615 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
617 EXPECT_EQ(GetBuiltinCode(model->operator_codes[1].get()), in TEST()
619 EXPECT_EQ(GetBuiltinCode(model->operator_codes[2].get()), in TEST()
/external/tensorflow/tensorflow/lite/schema/
Dschema_utils.h27 BuiltinOperator GetBuiltinCode(const OperatorCode *op_code);
29 BuiltinOperator GetBuiltinCode(const OperatorCodeT *op_code);
Dschema_utils.cc45 BuiltinOperator GetBuiltinCode(const OperatorCode* op_code) { in GetBuiltinCode() function
54 BuiltinOperator GetBuiltinCode(const OperatorCodeT* op_code) { in GetBuiltinCode() function
/external/tensorflow/tensorflow/lite/core/api/
Dop_resolver.cc30 auto builtin_code = GetBuiltinCode(opcode); in GetRegistrationFromOpCode()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc303 return tflite::EnumNamesBuiltinOperator()[GetBuiltinCode(&opcode)]; in GetOpName()
365 op_info.builtin_op_code = GetBuiltinCode(operator_code); in BuildLoggingInterpreter()
384 op_info.registration = op_resolver.FindOp(GetBuiltinCode(operator_code), in BuildLoggingInterpreter()
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops_no_kernel.cc44 if (tflite::GetBuiltinCode(opcode) != tflite::BuiltinOperator_CUSTOM || in AddFlexOpsFromModel()
Dgen_op_registration.cc41 auto builtin_code = GetBuiltinCode(opcode); in ReadOpsFromModel()
Dlist_flex_ops.cc91 if (tflite::GetBuiltinCode(opcode) != tflite::BuiltinOperator_CUSTOM || in AddFlexOpsFromModel()
Dverifier.cc538 auto builtin_code = GetBuiltinCode(opcode); in VerifySubGraphConsistency()
666 auto builtin_code = GetBuiltinCode(opcode); in VerifyOps()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dimporter_test_min_max.cc118 if (tflite::GetBuiltinCode( in InjectStatsToFullyConnected()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc175 auto builtin_code = GetBuiltinCode(opcode); in ExportAndSummarizeOperators()
665 GetBuiltinCode((*operator_codes)[0])); in TEST_F()
668 GetBuiltinCode((*operator_codes)[1])); in TEST_F()
Dimport.cc46 auto builtin_code = GetBuiltinCode(opcode); in LoadOperatorsTable()
/external/tensorflow/tensorflow/lite/tools/versioning/
Druntime_version.cc370 GetBuiltinCode(op_code), op_code->version()); in UpdateMinimumRuntimeVersionForModel()
Dop_version.cc676 auto builtin_code = GetBuiltinCode(op_code); in GetOpSignature()
860 auto builtin_code = GetBuiltinCode(op_code); in UpdateOpVersion()
/external/tensorflow/tensorflow/lite/
Dinterpreter_builder.cc221 if (GetBuiltinCode(opcode) == BuiltinOperator_CUSTOM) { in BuildLocalIndexToRegistrationMapping()
231 if (GetBuiltinCode(opcode) != BuiltinOperator_CUSTOM) { in BuildLocalIndexToRegistrationMapping()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc296 auto builtin_code = tflite::GetBuiltinCode(&op_code); in GetMlirOpName()
752 auto builtin_code = tflite::GetBuiltinCode(&op_code); in ConvertOp()

12