Home
last modified time | relevance | path

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

1234567

/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/tensorflow/tensorflow/lite/tools/
Dverifier.cc530 auto builtin_code = GetBuiltinCode(opcode); in VerifySubGraphConsistency() local
541 input_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
551 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
557 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
564 EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
571 EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
679 auto builtin_code = GetBuiltinCode(opcode); in VerifyOps() local
680 if (builtin_code < BuiltinOperator_MIN || in VerifyOps()
681 builtin_code > BuiltinOperator_MAX) { in VerifyOps()
683 builtin_code); in VerifyOps()
[all …]
Dgen_op_registration.cc41 auto builtin_code = GetBuiltinCode(opcode); in ReadOpsFromModel() local
42 if (builtin_code != ::tflite::BuiltinOperator_CUSTOM) { in ReadOpsFromModel()
44 std::make_pair(tflite::EnumNameBuiltinOperator(builtin_code), in ReadOpsFromModel()
/external/tensorflow/tensorflow/lite/
Dutil_test.cc99 registration.builtin_code = tflite::BuiltinOperator_ADD; in TEST()
106 registration.builtin_code = -1; in TEST()
113 registration.builtin_code = tflite::BuiltinOperator_CUSTOM; in TEST()
118 registration.builtin_code = tflite::BuiltinOperator_DELEGATE; in TEST()
Dmutable_op_resolver.cc69 new_registration.builtin_code = op; in AddBuiltin()
93 new_registration.builtin_code = BuiltinOperator_CUSTOM; in AddCustom()
/external/tensorflow/tensorflow/lite/python/
Dtest_util.py36 builtin_code = schema_util.get_builtin_code_from_operator_code(opcode)
37 if builtin_code == schema_fb.BuiltinOperator.CUSTOM:
41 op_set.add(visualize.BuiltinCodeToName(builtin_code))
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_signature.cc114 auto builtin_code = GetBuiltinCode(op_code); in GetOpSignature() local
115 OpSignature op_sig = {builtin_code}; in GetOpSignature()
118 if (builtin_code != BuiltinOperator_CUSTOM) { in GetOpSignature()
121 ParseOpData(op, builtin_code, &error_reporter, &allocator, in GetOpSignature()
127 switch (builtin_code) { in GetOpSignature()
221 static_cast<BuiltinOperator>(registration->builtin_code)}; in GetOpSignature()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc233 bool IsScalarInputSupported(int builtin_code) { in IsScalarInputSupported() argument
234 switch (builtin_code) { in IsScalarInputSupported()
258 bool NeedInt8Conversion(const TfLiteContext* context, int builtin_code, in NeedInt8Conversion() argument
262 switch (builtin_code) { in NeedInt8Conversion()
359 bool IsHybridOperator(const TfLiteContext* context, int builtin_code, in IsHybridOperator() argument
361 switch (builtin_code) { in IsHybridOperator()
409 return registration->builtin_code == kTfLiteBuiltinDequantize && in IsDequantizeConstFloat16()
417 return registration->builtin_code == kTfLiteBuiltinDequantize && in IsDequantizeNonConstFloat16()
425 return registration->builtin_code == kTfLiteBuiltinDensify && in IsDensifyConstTensor()
1883 const TfLiteContext* context, int builtin_code, int version, in Validate() argument
[all …]
Dnnapi_delegate_kernel.h286 static TfLiteStatus Map(TfLiteContext* context, int builtin_code, int version,
293 const TfLiteContext* context, int builtin_code, int version,
384 const TfLiteContext* context, int builtin_code, const TfLiteNode* node,
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc113 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildAddSubgraph()
144 mul_reg->builtin_code = kTfLiteBuiltinMul; in BuildMulSubgraph()
174 pad_reg->builtin_code = kTfLiteBuiltinPad; in BuildPadSubgraph()
208 if_reg->builtin_code = kTfLiteBuiltinIf; in BuildIfSubgraph()
240 le_reg->builtin_code = kTfLiteBuiltinLessEqual; in BuildLessEqualCondSubgraph()
284 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildAccumulateLoopBodySubgraph()
335 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildPadLoopBodySubgraph()
342 pad_reg->builtin_code = kTfLiteBuiltinPad; in BuildPadLoopBodySubgraph()
376 while_reg->builtin_code = kTfLiteBuiltinWhile; in BuildWhileSubgraph()
516 le_reg->builtin_code = kTfLiteBuiltinLessEqual; in BuildLessEqualCondSubgraphWithDynamicTensor()
[all …]
/external/armnn/src/armnnTfLiteParser/test/
DLoadScopeDynamicTensor.cpp26 "builtin_code": "AVERAGE_POOL_2D", in LoadScopeDynamicTensorFixture()
30 "builtin_code": "SOFTMAX", in LoadScopeDynamicTensorFixture()
187 "builtin_code": "ADD", in LoadScopeDynamicTensorBroadcastingFixture()
191 "builtin_code": "SUB", in LoadScopeDynamicTensorBroadcastingFixture()
DMinimum.cpp24 "operator_codes": [ { "builtin_code": "MINIMUM" } ], in MinimumFixture()
183 "operator_codes": [ { "builtin_code": "MINIMUM" } ], in MinimumBroadcastFixture2D0D()
DInputOutputTensorNames.cpp95 "operator_codes": [ { "builtin_code": "AVERAGE_POOL_2D" } ], in ValidTensorsFixture()
156 "operator_codes": [ { "builtin_code": "MINIMUM" } ], in Rank0TensorFixture()
DLoadModel.cpp25 … "operator_codes": [ { "builtin_code": "AVERAGE_POOL_2D" }, { "builtin_code": "CONV_2D" } ], in LoadModelFixture()
155 BOOST_CHECK_EQUAL(expectedOperators[i], result[i]->builtin_code); in CheckBuiltinOperators()
/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
Dflex_op_with_tflite_op.mlir10 // CHECK-NEXT: builtin_code: MUL
14 // CHECK-NEXT: builtin_code: CUSTOM
18 // CHECK-NEXT: builtin_code: EXP
Dcustom_op_with_tflite_op.mlir11 // CHECK-NEXT: builtin_code: MUL
15 // CHECK-NEXT: builtin_code: CUSTOM
19 // CHECK-NEXT: builtin_code: EXP
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc47 reg.builtin_code = tflite::BuiltinOperator_CUSTOM; in AddOpRegistration()
170 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
173 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
187 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
190 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
417 add_reg->builtin_code = kTfLiteBuiltinAdd; in SetUp()
418 deq_reg->builtin_code = kTfLiteBuiltinDequantize; in SetUp()
419 mul_reg->builtin_code = kTfLiteBuiltinMul; in SetUp()
476 return registration->builtin_code == kTfLiteBuiltinAdd; in FP16Delegate()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dlogging_op_resolver_test.cc77 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CONV_2D); in TEST()
83 EXPECT_EQ(reg->builtin_code, BuiltinOperator_ADD); in TEST()
136 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CONV_2D); in TEST()
162 EXPECT_EQ(reg->builtin_code, BuiltinOperator_CUSTOM); in TEST()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc305 if (registration->builtin_code == kTfLiteBuiltinConv2d) { in IsConvolutionOpSupported()
309 } else if (registration->builtin_code == kTfLiteBuiltinDepthwiseConv2d) { in IsConvolutionOpSupported()
313 } else if (registration->builtin_code == kTfLiteBuiltinTransposeConv) { in IsConvolutionOpSupported()
340 if (registration->builtin_code == kTfLiteBuiltinTransposeConv) { in IsConvolutionOpSupported()
/external/tensorflow/tensorflow/lite/python/analyzer_wrapper/
Dmodel_analyzer.cc86 auto builtin_code = GetBuiltinCode(op_code); in get_op_name() local
87 if (builtin_code != BuiltinOperator_CUSTOM) { in get_op_name()
88 return EnumNameBuiltinOperator(builtin_code); in get_op_name()

1234567