Home
last modified time | relevance | path

Searched refs:opcode_index (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/optimize/
Dmodify_model_interface_test.cc62 quant_op->opcode_index = 0; in CreateQuantizedModelSingleInputOutput()
66 fc_op->opcode_index = 1; in CreateQuantizedModelSingleInputOutput()
70 dequant_op->opcode_index = 2; in CreateQuantizedModelSingleInputOutput()
159 quant_op_1->opcode_index = 0; in CreateQuantizedModelMultipleInputOutput()
162 quant_op_2->opcode_index = 0; in CreateQuantizedModelMultipleInputOutput()
166 fc_op->opcode_index = 1; in CreateQuantizedModelMultipleInputOutput()
170 dequant_op_1->opcode_index = 2; in CreateQuantizedModelMultipleInputOutput()
173 dequant_op_2->opcode_index = 2; in CreateQuantizedModelMultipleInputOutput()
276 fc_op->opcode_index = 0; in CreateFloatModel()
331 EXPECT_EQ(model->subgraphs[0]->operators[0]->opcode_index, 1); in TEST_P()
[all …]
Dquantize_model_test.cc231 EXPECT_EQ(quant_op->opcode_index, float_op->opcode_index()); in TEST_P()
274 const int32_t quant_idx = quant_op->opcode_index; in TEST_P()
275 const int32_t dequant_idx = dequant_op->opcode_index; in TEST_P()
339 const int32_t quant_op_uint8_int8_idx = quant_op_uint8_int8->opcode_index; in TEST_P()
340 const int32_t quant_op_int8_uint8_idx = quant_op_int8_uint8->opcode_index; 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()
553 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[split->opcode_index].get()), in TEST_F()
555 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[add->opcode_index].get()), in TEST_F()
834 ASSERT_EQ(op->opcode_index, 0); in TEST_F()
[all …]
Dquantize_weights_test.cc124 const uint32_t op_code_idx = op->opcode_index(); in GetProducerOpCode()
199 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
258 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
317 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
368 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
402 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
442 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
482 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
528 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
564 const uint32_t op_code_idx = op->opcode_index(); in TEST_F()
Dmodel_utils.cc63 op_raw->opcode_index = in MakeDequantizeOperator()
75 op_raw->opcode_index = in MakeQuantizeOperator()
139 OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get(); in SetOperatorCodeVersion()
Dquantize_weights.cc201 const OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get(); in InsertQuantizableInputTensorsFromOperator()
319 op_raw->opcode_index = GetOrInsertDequantizeOpCodeIndex(model); in MakeDequantizeOperator()
377 GetBuiltinCode(model->operator_codes[consumer_op->opcode_index].get()); in IsQuantizationPassThroughOps()
393 const OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get(); in PassQuantizationAndGetConsumers()
474 model->operator_codes[consumer_op->opcode_index].get(); in QuantizeWeightsInt8()
Dquantize_model.cc63 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in GetOperatorProperty()
231 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in InputQuantizeRequired()
530 OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get(); in SetOperatorPropertyADDSUBOperator()
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()
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()
241 op->opcode_index = quant_op_index; in SetOutputTypeToUINT8()
Dquantization_wrapper_utils_test.cc42 lstm_op->opcode_index = 0; in TEST()
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()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dmath.mlir85 // CHECK-NEXT: opcode_index: 1,
93 // CHECK-NEXT: opcode_index: 2,
101 // CHECK-NEXT: opcode_index: 3,
109 // CHECK-NEXT: opcode_index: 4,
Dquantization.mlir111 // CHECK-NEXT: opcode_index: 1,
122 // CHECK-NEXT: opcode_index: 2,
126 // CHECK-NEXT: opcode_index: 3,
134 // CHECK-NEXT: opcode_index: 4,
Dif_op.mlir58 // CHECK-NEXT: opcode_index: 1,
94 // CHECK-NEXT: opcode_index: 2,
129 // CHECK-NEXT: opcode_index: 3,
Dcustom_op_with_tflite_op.mlir68 // CHECK-NEXT: opcode_index: 1,
73 // CHECK-NEXT: opcode_index: 2,
Dflex_op_with_tflite_op.mlir67 // CHECK-NEXT: opcode_index: 1,
72 // CHECK-NEXT: opcode_index: 2,
Dwhile_op.mlir100 // CHECK-NEXT: opcode_index: 1,
148 // CHECK-NEXT: opcode_index: 2,
156 // CHECK-NEXT: opcode_index: 3,
Dtfl_while_op.mlir100 // CHECK-NEXT: opcode_index: 1,
148 // CHECK-NEXT: opcode_index: 2,
156 // CHECK-NEXT: opcode_index: 3,
Dlogical.mlir64 // CHECK-NEXT: opcode_index: 1,
Ddisable_flex_enable_builtin.mlir70 // CHECK-NEXT: opcode_index: 1,
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc388 const int opcode_index = opcodes_.size() - 1; in AddBuiltinOp() local
390 builder_, opcode_index, builder_.CreateVector<int32_t>(inputs), in AddBuiltinOp()
403 const int opcode_index = opcodes_.size() - 1; in AddCustomOp() local
405 builder_, opcode_index, builder_.CreateVector<int32_t>(inputs), in AddCustomOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc853 auto opcode_index = GetOpcodeIndex("if", tflite::BuiltinOperator_IF); in BuildIfOperator() local
861 return tflite::CreateOperator(builder_, opcode_index, inputs, outputs, in BuildIfOperator()
869 auto opcode_index = in BuildCallOnceOperator() local
877 return tflite::CreateOperator(builder_, opcode_index, inputs, outputs, in BuildCallOnceOperator()
885 auto opcode_index = GetOpcodeIndex("while", tflite::BuiltinOperator_WHILE); in BuildWhileOperator() local
893 return tflite::CreateOperator(builder_, opcode_index, inputs, outputs, in BuildWhileOperator()
901 auto opcode_index = GetOpcodeIndex("while", tflite::BuiltinOperator_WHILE); in BuildWhileOperator() local
919 return tflite::CreateOperator(builder_, opcode_index, inputs, outputs, in BuildWhileOperator()
937 auto opcode_index = in BuildNumericVerifyOperator() local
940 builder_, opcode_index, builder_.CreateVector(operands), in BuildNumericVerifyOperator()
[all …]
Dflatbuffer_operator.h46 Operation *mlir_op, uint32_t opcode_index,
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops_no_kernel.cc43 const tflite::OperatorCode* opcode = opcodes->Get(op->opcode_index()); in AddFlexOpsFromModel()
Dverifier.cc531 (op->opcode_index() >= model.operator_codes()->size())) { in VerifySubGraphConsistency()
534 op->opcode_index()); in VerifySubGraphConsistency()
537 const auto& opcode = model.operator_codes()->Get(op->opcode_index()); in VerifySubGraphConsistency()
Dlist_flex_ops.cc90 const tflite::OperatorCode* opcode = opcodes->Get(op->opcode_index()); in AddFlexOpsFromModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dimporter_test_min_max.cc119 model->operator_codes[op->opcode_index].get()) == in InjectStatsToFullyConnected()

12