| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_model_test.cc | 194 EXPECT_EQ(model_.operator_codes.size(), 1); in TEST_P()
195 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[0].get()), in TEST_P()
197 EXPECT_EQ(model_.operator_codes[0]->version, 3); in TEST_P()
205 ASSERT_EQ(model_.operator_codes.size(), in TEST_P()
206 readonly_model_->operator_codes()->size()); in TEST_P()
207 for (size_t i = 0; i < model_.operator_codes.size(); i++) { in TEST_P()
208 const auto float_model_op = readonly_model_->operator_codes()->Get(i); in TEST_P()
209 EXPECT_EQ(GetBuiltinCode(model_.operator_codes[i].get()), in TEST_P()
211 if (GetBuiltinCode(model_.operator_codes[i].get()) == in TEST_P()
213 EXPECT_EQ(model_.operator_codes[i]->version, 3); in TEST_P()
[all …]
|
| D | quantize_weights.cc | 201 const OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get(); in InsertQuantizableInputTensorsFromOperator()
297 for (size_t i = 0; i < model->operator_codes.size(); ++i) { in GetOrInsertDequantizeOpCodeIndex()
298 if (GetBuiltinCode(model->operator_codes[i].get()) == in GetOrInsertDequantizeOpCodeIndex()
303 model->operator_codes.push_back(absl::make_unique<OperatorCodeT>()); in GetOrInsertDequantizeOpCodeIndex()
304 int op_code_idx = model->operator_codes.size() - 1; in GetOrInsertDequantizeOpCodeIndex()
305 model->operator_codes[op_code_idx]->builtin_code = BuiltinOperator_DEQUANTIZE; in GetOrInsertDequantizeOpCodeIndex()
306 model->operator_codes[op_code_idx]->deprecated_builtin_code = in GetOrInsertDequantizeOpCodeIndex()
309 model->operator_codes[op_code_idx]->version = 2; in GetOrInsertDequantizeOpCodeIndex()
342 for (int i = 0, end = model->operator_codes.size(); i < end; ++i) { in UpdateInt8OperatorVersions()
344 GetBuiltinCode(model->operator_codes[i].get()); in UpdateInt8OperatorVersions()
[all …]
|
| D | model_utils.cc | 39 for (size_t i = 0; i < model->operator_codes.size(); ++i) { in GetOrInsertOpCodeIndex()
40 if (GetBuiltinCode(model->operator_codes[i].get()) == op_code) { in GetOrInsertOpCodeIndex()
44 model->operator_codes.push_back(absl::make_unique<OperatorCodeT>()); in GetOrInsertOpCodeIndex()
45 int op_code_idx = model->operator_codes.size() - 1; in GetOrInsertOpCodeIndex()
46 model->operator_codes[op_code_idx]->builtin_code = op_code; in GetOrInsertOpCodeIndex()
47 model->operator_codes[op_code_idx]->deprecated_builtin_code = in GetOrInsertOpCodeIndex()
50 model->operator_codes[op_code_idx]->version = version; in GetOrInsertOpCodeIndex()
139 OperatorCodeT* op_code = model->operator_codes[op->opcode_index].get(); in SetOperatorCodeVersion()
|
| D | quantization_wrapper_utils_test.cc | 60 model->operator_codes.push_back(std::move(lstm_op_code)); in TEST()
68 EXPECT_EQ(model->operator_codes.size(), 1); in TEST()
74 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
95 EXPECT_EQ(model->operator_codes.size(), 1); in TEST()
101 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
|
| D | quantization_wrapper_utils_custom_test.cc | 59 model->operator_codes.push_back(std::move(lstm_op_code)); in TEST()
67 EXPECT_EQ(model->operator_codes.size(), 1); in TEST()
73 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
102 EXPECT_EQ(model->operator_codes.size(), 1); in TEST()
108 EXPECT_EQ(GetBuiltinCode(model->operator_codes[0].get()), in TEST()
|
| D | modify_model_interface_test.cc | 78 model->operator_codes.push_back(std::move(quant_op_code)); in CreateQuantizedModelSingleInputOutput()
79 model->operator_codes.push_back(std::move(fc_op_code)); in CreateQuantizedModelSingleInputOutput()
80 model->operator_codes.push_back(std::move(dequant_op_code)); in CreateQuantizedModelSingleInputOutput()
183 model->operator_codes.push_back(std::move(quant_op_code)); in CreateQuantizedModelMultipleInputOutput()
184 model->operator_codes.push_back(std::move(fc_op_code)); in CreateQuantizedModelMultipleInputOutput()
185 model->operator_codes.push_back(std::move(dequant_op_code)); in CreateQuantizedModelMultipleInputOutput()
281 model->operator_codes.push_back(std::move(fc_op_code)); in CreateFloatModel()
329 EXPECT_EQ(model->operator_codes.size(), 3); in TEST_P()
370 EXPECT_EQ(model->operator_codes.size(), 3); in TEST_P()
420 EXPECT_EQ(model->operator_codes.size(), 3); in TEST()
[all …]
|
| D | quantize_weights_test.cc | 125 *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()
|
| D | modify_model_interface.cc | 69 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in GetInputTensors()
144 GetBuiltinCode(model->operator_codes[op->opcode_index].get()); in GetOutputTensors()
219 for (size_t i = 0; i < model->operator_codes.size(); ++i) { in SetOutputTypeToUINT8()
220 if (GetBuiltinCode(model->operator_codes[i].get()) == in SetOutputTypeToUINT8()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | conv_2d_tester.cc | 122 std::vector<flatbuffers::Offset<OperatorCode>> operator_codes{ in CreateTfLiteModel() local
129 operator_codes.emplace_back( in CreateTfLiteModel()
135 builder, /*opcode_index=*/operator_codes.size() - 1, in CreateTfLiteModel()
143 operator_codes.emplace_back( in CreateTfLiteModel()
188 builder, /*opcode_index=*/operator_codes.size() - 1, in CreateTfLiteModel()
198 builder, /*opcode_index=*/operator_codes.size() - 1, in CreateTfLiteModel()
335 builder.CreateVector(operator_codes.data(), operator_codes.size()), in CreateTfLiteModel()
|
| D | prelu_tester.cc | 107 std::vector<flatbuffers::Offset<OperatorCode>> operator_codes{ in CreateTfLiteModel() local
110 operator_codes.emplace_back( in CreateTfLiteModel()
113 operator_codes.emplace_back( in CreateTfLiteModel()
226 builder.CreateVector(operator_codes.data(), operator_codes.size()), in CreateTfLiteModel()
|
| D | depthwise_conv_2d_tester.cc | 122 std::vector<flatbuffers::Offset<OperatorCode>> operator_codes{ in CreateTfLiteModel() local
129 operator_codes.emplace_back( in CreateTfLiteModel()
221 operator_codes.emplace_back( in CreateTfLiteModel()
327 builder.CreateVector(operator_codes.data(), operator_codes.size()), in CreateTfLiteModel()
|
| D | fully_connected_tester.cc | 122 std::vector<flatbuffers::Offset<OperatorCode>> operator_codes{ in CreateTfLiteModel() local
129 operator_codes.emplace_back( in CreateTfLiteModel()
276 builder.CreateVector(operator_codes.data(), operator_codes.size()), in CreateTfLiteModel()
|
| D | binary_elementwise_tester.cc | 192 std::vector<flatbuffers::Offset<OperatorCode>> operator_codes{ in CreateTfLiteModel() local
195 operator_codes.emplace_back( in CreateTfLiteModel()
198 operator_codes.emplace_back( in CreateTfLiteModel()
399 builder.CreateVector(operator_codes.data(), operator_codes.size()), in CreateTfLiteModel()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export_test.cc | 174 for (const ::tflite::OperatorCode* opcode : *model->operator_codes()) { in ExportAndSummarizeOperators()
659 auto operator_codes = model->operator_codes(); in TEST_F() local
663 EXPECT_EQ(2, operator_codes->size()); in TEST_F()
665 GetBuiltinCode((*operator_codes)[0])); in TEST_F()
666 EXPECT_EQ(1, (*operator_codes)[0]->version()); in TEST_F()
668 GetBuiltinCode((*operator_codes)[1])); in TEST_F()
669 EXPECT_EQ(2, (*operator_codes)[1]->version()); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/testdata/ |
| D | add.json | 3 operator_codes: [
|
| D | add_quantized_int8.json | 3 operator_codes: [
|
| D | add_quantized.json | 3 operator_codes: [
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 530 if (!model.operator_codes() || in VerifySubGraphConsistency()
531 (op->opcode_index() >= model.operator_codes()->size())) { in VerifySubGraphConsistency()
537 const auto& opcode = model.operator_codes()->Get(op->opcode_index()); in VerifySubGraphConsistency()
662 if (!model.operator_codes()) { in VerifyOps()
665 for (const auto& opcode : *model.operator_codes()) { in VerifyOps()
|
| D | list_flex_ops_no_kernel.cc | 38 auto* opcodes = model->operator_codes(); in AddFlexOpsFromModel()
|
| D | gen_op_registration.cc | 37 auto opcodes = model->operator_codes(); in ReadOpsFromModel()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 353 auto operator_codes = tflite_model->operator_codes(); in BuildLoggingInterpreter() local
364 auto operator_code = operator_codes->Get(op->opcode_index()); in BuildLoggingInterpreter()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | import_json.json | 9 operator_codes: [
|
| D | optional_input.json | 11 operator_codes: [
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | hashtable_resource.mlir | 5 // CHECK: operator_codes: [ {
|
| D | type_attr.mlir | 5 // CHECK: operator_codes: [ {
|