Home
last modified time | relevance | path

Searched refs:builtin_code (Results 1 – 25 of 200) sorted by relevance

12345678

/aosp_15_r20/external/tensorflow/tensorflow/lite/schema/
H A Dschema_conversion_utils.cc24 const BuiltinOperator builtin_code) { in ConvertBuiltinCodeToDeprecatedBuiltinCode() argument
25 return (builtin_code < BuiltinOperator_PLACEHOLDER_FOR_GREATER_OP_CODES) in ConvertBuiltinCodeToDeprecatedBuiltinCode()
26 ? static_cast<int8_t>(builtin_code) in ConvertBuiltinCodeToDeprecatedBuiltinCode()
41 flatbuffers::FlatBufferBuilder &_fbb, BuiltinOperator builtin_code, in CreateOperatorCode() argument
48 if (builtin_code < BuiltinOperator_PLACEHOLDER_FOR_GREATER_OP_CODES) { in CreateOperatorCode()
49 deprecated_builtin_code = static_cast<int8_t>(builtin_code); in CreateOperatorCode()
53 builder_.add_builtin_code(builtin_code); in CreateOperatorCode()
58 flatbuffers::FlatBufferBuilder &_fbb, BuiltinOperator builtin_code, in CreateOperatorCodeDirect() argument
63 if (builtin_code < BuiltinOperator_PLACEHOLDER_FOR_GREATER_OP_CODES) { in CreateOperatorCodeDirect()
64 deprecated_builtin_code = static_cast<int8_t>(builtin_code); in CreateOperatorCodeDirect()
[all …]
H A Dschema_conversion_utils.h24 const BuiltinOperator builtin_code);
30 BuiltinOperator builtin_code = BuiltinOperator_ADD,
36 BuiltinOperator builtin_code = BuiltinOperator_ADD,
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/api/
H A Dop_resolver.cc30 auto builtin_code = GetBuiltinCode(opcode); in GetRegistrationFromOpCode() local
33 if (builtin_code > BuiltinOperator_MAX) { in GetRegistrationFromOpCode()
38 builtin_code); in GetRegistrationFromOpCode()
40 } else if (builtin_code != BuiltinOperator_CUSTOM) { in GetRegistrationFromOpCode()
41 *registration = op_resolver.FindOp(builtin_code, version); in GetRegistrationFromOpCode()
48 EnumNameBuiltinOperator(builtin_code), version); in GetRegistrationFromOpCode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier.cc534 auto builtin_code = GetBuiltinCode(opcode); in VerifySubGraphConsistency() local
545 input_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
555 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
561 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
568 EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
575 EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
683 auto builtin_code = GetBuiltinCode(opcode); in VerifyOps() local
684 if (builtin_code < BuiltinOperator_MIN || in VerifyOps()
685 builtin_code > BuiltinOperator_MAX) { in VerifyOps()
687 builtin_code); in VerifyOps()
[all …]
H A Dgen_op_registration.cc42 auto builtin_code = GetBuiltinCode(opcode); in ReadOpsFromModel() local
43 if (builtin_code != ::tflite::BuiltinOperator_CUSTOM) { in ReadOpsFromModel()
45 std::make_pair(tflite::EnumNameBuiltinOperator(builtin_code), in ReadOpsFromModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc236 bool IsScalarInputSupported(int builtin_code) { in IsScalarInputSupported() argument
237 switch (builtin_code) { in IsScalarInputSupported()
261 bool NeedInt8Conversion(const TfLiteContext* context, int builtin_code, in NeedInt8Conversion() argument
265 switch (builtin_code) { in NeedInt8Conversion()
385 bool IsHybridOperator(const TfLiteContext* context, int builtin_code, in IsHybridOperator() argument
387 switch (builtin_code) { in IsHybridOperator()
435 return registration->builtin_code == kTfLiteBuiltinDequantize && in IsDequantizeConstFloat16()
443 return registration->builtin_code == kTfLiteBuiltinDequantize && in IsDequantizeNonConstFloat16()
451 return registration->builtin_code == kTfLiteBuiltinDensify && in IsDensifyConstTensor()
2300 auto builtin_code = registration->builtin_code; in Validate() local
[all …]
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/
H A DReshapeDynamic.cpp22 "builtin_code": "ADD" in ReshapeDynamicFixture1()
27 "builtin_code": "ADD" in ReshapeDynamicFixture1()
187 "builtin_code": "ADD" in ReshapeDynamicFixture2()
192 "builtin_code": "ADD" in ReshapeDynamicFixture2()
350 "builtin_code": "ADD" in ReshapeDynamicFixture3()
355 "builtin_code": "ADD" in ReshapeDynamicFixture3()
H A DPrelu.cpp25 "builtin_code": "PRELU", in PreluFixture()
113 "builtin_code": "PRELU", in PreluNetworkFixture()
117 "builtin_code": "MUL", in PreluNetworkFixture()
121 "builtin_code": "ADD", in PreluNetworkFixture()
H A DLoadScopeDynamicTensor.cpp23 "builtin_code": "AVERAGE_POOL_2D", in LoadScopeDynamicTensorFixture()
27 "builtin_code": "SOFTMAX", in LoadScopeDynamicTensorFixture()
184 "builtin_code": "ADD", in LoadScopeDynamicTensorBroadcastingFixture()
188 "builtin_code": "SUB", in LoadScopeDynamicTensorBroadcastingFixture()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsubgraph_test_util.cc114 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildAddSubgraph()
145 mul_reg->builtin_code = kTfLiteBuiltinMul; in BuildMulSubgraph()
175 pad_reg->builtin_code = kTfLiteBuiltinPad; in BuildPadSubgraph()
209 if_reg->builtin_code = kTfLiteBuiltinIf; in BuildIfSubgraph()
241 le_reg->builtin_code = kTfLiteBuiltinLessEqual; in BuildLessEqualCondSubgraph()
285 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildAccumulateLoopBodySubgraph()
336 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildPadLoopBodySubgraph()
343 pad_reg->builtin_code = kTfLiteBuiltinPad; in BuildPadLoopBodySubgraph()
377 while_reg->builtin_code = kTfLiteBuiltinWhile; in BuildWhileSubgraph()
517 le_reg->builtin_code = kTfLiteBuiltinLessEqual; in BuildLessEqualCondSubgraphWithDynamicTensor()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Ddelegate_test_util.cc47 reg.builtin_code = tflite::BuiltinOperator_CUSTOM; in AddOpRegistration()
175 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
178 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
192 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
195 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
425 add_reg->builtin_code = kTfLiteBuiltinAdd; in SetUp()
426 deq_reg->builtin_code = kTfLiteBuiltinDequantize; in SetUp()
427 mul_reg->builtin_code = kTfLiteBuiltinMul; in SetUp()
484 return registration->builtin_code == kTfLiteBuiltinAdd; in FP16Delegate()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/
H A Dop_signature.cc126 auto builtin_code = GetBuiltinCode(op_code); in GetOpSignature() local
127 OpSignature op_sig = {builtin_code}; in GetOpSignature()
130 if (builtin_code != BuiltinOperator_CUSTOM) { in GetOpSignature()
133 ParseOpData(op, builtin_code, &error_reporter, &allocator, in GetOpSignature()
139 switch (builtin_code) { in GetOpSignature()
258 static_cast<BuiltinOperator>(registration->builtin_code)}; in GetOpSignature()
H A Dop_signature_test.cc33 StubTfLiteContext(const int builtin_code, const int op_version, in StubTfLiteContext() argument
83 registrations_[0].builtin_code = kTfLiteBuiltinAdd;
84 registrations_[1].builtin_code = builtin_code;
86 registrations_[2].builtin_code = kTfLiteBuiltinAdd;
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dutil_test.cc100 registration.builtin_code = tflite::BuiltinOperator_ADD; in TEST()
107 registration.builtin_code = -1; in TEST()
114 registration.builtin_code = tflite::BuiltinOperator_CUSTOM; in TEST()
119 registration.builtin_code = tflite::BuiltinOperator_DELEGATE; in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
H A Dmath.mlir10 // CHECK-NEXT: builtin_code: SQUARED_DIFFERENCE
14 // CHECK-NEXT: builtin_code: MUL
18 // CHECK-NEXT: builtin_code: DIV
22 // CHECK-NEXT: builtin_code: EXP
26 // CHECK-NEXT: builtin_code: NEG
H A Dquantization.mlir9 // CHECK-NEXT: builtin_code: QUANTIZE
13 // CHECK-NEXT: builtin_code: CONV_2D
17 // CHECK-NEXT: builtin_code: RESHAPE
21 // CHECK-NEXT: builtin_code: SOFTMAX
25 // CHECK-NEXT: builtin_code: DEQUANTIZE
H A Dcustom_op_with_tflite_op.mlir11 // CHECK-NEXT: builtin_code: MUL
15 // CHECK-NEXT: builtin_code: CUSTOM
19 // CHECK-NEXT: builtin_code: EXP
H A Dflex_op_with_tflite_op.mlir10 // CHECK-NEXT: builtin_code: MUL
14 // CHECK-NEXT: builtin_code: CUSTOM
18 // CHECK-NEXT: builtin_code: EXP
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_api_test.cc402 EXPECT_EQ(reg_add->builtin_code, kTfLiteBuiltinAdd); in TEST()
413 EXPECT_EQ(reg_sinh->builtin_code, kTfLiteBuiltinCustom); in TEST()
455 EXPECT_EQ(reg_add->builtin_code, kTfLiteBuiltinAdd); in TEST()
469 EXPECT_EQ(reg_add->builtin_code, kTfLiteBuiltinAdd); in TEST()
476 EXPECT_EQ(reg_sinh->builtin_code, kTfLiteBuiltinCustom); in TEST()
487 EXPECT_EQ(reg_sinh->builtin_code, kTfLiteBuiltinCustom); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A Dlogging_op_resolver_test.cc79 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CONV_2D); in TEST()
85 EXPECT_EQ(reg->builtin_code, BuiltinOperator_ADD); in TEST()
138 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CONV_2D); in TEST()
164 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CUSTOM); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dtest_util.py32 builtin_code = schema_util.get_builtin_code_from_operator_code(opcode)
33 if builtin_code == schema_fb.BuiltinOperator.CUSTOM:
37 op_set.add(visualize.BuiltinCodeToName(builtin_code))
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_operator.cc67 auto builtin_code = tflite::GetBuiltinCode(&op_code); in GetMlirOpNameFromOpCode() local
68 if (builtin_code == tflite::BuiltinOperator_CUSTOM) { in GetMlirOpNameFromOpCode()
71 if (builtin_code == tflite::BuiltinOperator_IF) { in GetMlirOpNameFromOpCode()
74 if (builtin_code == tflite::BuiltinOperator_WHILE) { in GetMlirOpNameFromOpCode()
78 llvm::StringRef op_name(tflite::EnumNameBuiltinOperator(builtin_code)); in GetMlirOpNameFromOpCode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dconvolution_op_builder.cc309 if (registration->builtin_code == kTfLiteBuiltinConv2d) { in IsConvolutionOpSupported()
313 } else if (registration->builtin_code == kTfLiteBuiltinDepthwiseConv2d) { in IsConvolutionOpSupported()
317 } else if (registration->builtin_code == kTfLiteBuiltinTransposeConv) { in IsConvolutionOpSupported()
344 if (registration->builtin_code == kTfLiteBuiltinTransposeConv) { in IsConvolutionOpSupported()
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc85 registration.builtin_code)), in ReportOpError()
162 if (op_reg.builtin_code == tflite::BuiltinOperator_CUSTOM) { in GetTFLiteOpName()
166 if (op_reg.builtin_code == tflite::BuiltinOperator_DELEGATE && in GetTFLiteOpName()
170 return tflite::EnumNamesBuiltinOperator()[op_reg.builtin_code]; in GetTFLiteOpName()
411 registration.builtin_code = BuiltinOperator_DELEGATE; in ReplaceNodeSubsetsWithDelegateKernels()
843 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in AddNodeWithParameters()
885 if (registration->builtin_code == kTfLiteBuiltinIf || in OpMightHaveSideEffect()
886 registration->builtin_code == kTfLiteBuiltinWhile || in OpMightHaveSideEffect()
887 registration->builtin_code == kTfLiteBuiltinCallOnce) in OpMightHaveSideEffect()
1286 if (registration.builtin_code == kTfLiteBuiltinReshape && i == 1 && in Invoke()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dmodify_model_interface_test.cc47 quant_op_code->builtin_code = BuiltinOperator_QUANTIZE; in CreateQuantizedModelSingleInputOutput()
52 fc_op_code->builtin_code = BuiltinOperator_FULLY_CONNECTED; in CreateQuantizedModelSingleInputOutput()
57 dequant_op_code->builtin_code = BuiltinOperator_DEQUANTIZE; in CreateQuantizedModelSingleInputOutput()
144 quant_op_code->builtin_code = BuiltinOperator_QUANTIZE; in CreateQuantizedModelMultipleInputOutput()
149 fc_op_code->builtin_code = BuiltinOperator_FULLY_CONNECTED; in CreateQuantizedModelMultipleInputOutput()
154 dequant_op_code->builtin_code = BuiltinOperator_DEQUANTIZE; in CreateQuantizedModelMultipleInputOutput()
271 fc_op_code->builtin_code = BuiltinOperator_FULLY_CONNECTED; in CreateFloatModel()

12345678