Home
last modified time | relevance | path

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

12345678

/external/tensorflow/tensorflow/lite/schema/
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 …]
Dschema_conversion_utils.h24 const BuiltinOperator builtin_code);
30 BuiltinOperator builtin_code = BuiltinOperator_ADD,
36 BuiltinOperator builtin_code = BuiltinOperator_ADD,
Dschema_utils.cc50 op_code->builtin_code(), in GetBuiltinCode()
58 return std::max(op_code->builtin_code, static_cast<BuiltinOperator>( in GetBuiltinCode()
/external/tensorflow/tensorflow/lite/core/api/
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()
/external/armnn/src/armnnTfLiteParser/test/
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()
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()
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()
DTransposeConv.cpp23 "operator_codes": [ { "builtin_code": "TRANSPOSE_CONV" } ], in TransposeConvFixture()
147 "operator_codes": [ { "builtin_code": "TRANSPOSE_CONV" } ], in TransposeConvFixtureWithBias()
281 "builtin_code": "TRANSPOSE_CONV", in TransposeConvPerChannelFixture()
/external/tensorflow/tensorflow/lite/tools/
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 …]
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()
/external/tensorflow/tensorflow/lite/python/
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))
/external/tensorflow/tensorflow/lite/
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()
/external/tensorflow/tensorflow/lite/kernels/
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 …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
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 …]
/external/tensorflow/tensorflow/lite/tools/versioning/
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()
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;
/external/tensorflow/tensorflow/lite/delegates/
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()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
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
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
Dcustom_op_with_tflite_op.mlir11 // CHECK-NEXT: builtin_code: MUL
15 // CHECK-NEXT: builtin_code: CUSTOM
19 // CHECK-NEXT: builtin_code: EXP
Dflex_op_with_tflite_op.mlir10 // CHECK-NEXT: builtin_code: MUL
14 // CHECK-NEXT: builtin_code: CUSTOM
18 // CHECK-NEXT: builtin_code: EXP
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
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()
/external/tensorflow/tensorflow/compiler/mlir/lite/
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()
/external/tensorflow/tensorflow/lite/c/
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()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
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()

12345678