Home
last modified time | relevance | path

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

12345

/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()
34 builtin_code < BuiltinOperator_MIN) { in GetRegistrationFromOpCode()
39 builtin_code); in GetRegistrationFromOpCode()
41 } else if (builtin_code != BuiltinOperator_CUSTOM) { in GetRegistrationFromOpCode()
42 *registration = op_resolver.FindOp(builtin_code, version); in GetRegistrationFromOpCode()
49 EnumNameBuiltinOperator(builtin_code), version); in GetRegistrationFromOpCode()
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc538 auto builtin_code = GetBuiltinCode(opcode); in VerifySubGraphConsistency() local
549 input_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
559 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
565 output_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
572 EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
579 EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
666 auto builtin_code = GetBuiltinCode(opcode); in VerifyOps() local
667 if (builtin_code < BuiltinOperator_MIN || in VerifyOps()
668 builtin_code > BuiltinOperator_MAX) { in VerifyOps()
670 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.cc44 new_registration.builtin_code = op; in AddBuiltin()
62 new_registration.builtin_code = BuiltinOperator_CUSTOM; in AddCustom()
Doptional_debug_tools.cc136 reg.builtin_code, EnumNamesBuiltinOperator()[reg.builtin_code]); in PrintInterpreterState()
Dmutable_op_resolver_test.cc75 EXPECT_EQ(found_registration->builtin_code, BuiltinOperator_ADD); in TEST()
145 EXPECT_EQ(found_registration->builtin_code, BuiltinOperator_CUSTOM); in TEST()
160 EXPECT_EQ(found_registration->builtin_code, BuiltinOperator_CUSTOM); in TEST()
Dutil.cc160 return registration.builtin_code == tflite::BuiltinOperator_CUSTOM && in IsUnresolvedCustomOp()
165 auto op = registration.builtin_code; in GetOpNameByRegistration()
/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/kernels/
Dsubgraph_test_util.cc119 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildAddSubgraph()
150 mul_reg->builtin_code = kTfLiteBuiltinMul; in BuildMulSubgraph()
180 pad_reg->builtin_code = kTfLiteBuiltinPad; in BuildPadSubgraph()
214 if_reg->builtin_code = kTfLiteBuiltinIf; in BuildIfSubgraph()
246 le_reg->builtin_code = kTfLiteBuiltinLessEqual; in BuildLessEqualCondSubgraph()
290 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildAccumulateLoopBodySubgraph()
341 add_reg->builtin_code = kTfLiteBuiltinAdd; in BuildPadLoopBodySubgraph()
348 pad_reg->builtin_code = kTfLiteBuiltinPad; in BuildPadLoopBodySubgraph()
382 while_reg->builtin_code = kTfLiteBuiltinWhile; in BuildWhileSubgraph()
478 le_reg->builtin_code = kTfLiteBuiltinLessEqual; in BuildLessEqualCondSubgraphWithDynamicTensor()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc210 bool IsScalarInputSupported(int builtin_code) { in IsScalarInputSupported() argument
211 switch (builtin_code) { in IsScalarInputSupported()
235 bool NeedInt8Conversion(const TfLiteContext* context, int builtin_code, in NeedInt8Conversion() argument
239 switch (builtin_code) { in NeedInt8Conversion()
336 bool IsHybridOperator(const TfLiteContext* context, int builtin_code, in IsHybridOperator() argument
338 switch (builtin_code) { in IsHybridOperator()
1680 const TfLiteContext* context, int builtin_code, int version, in Validate() argument
1685 switch (builtin_code) { in Validate()
1727 if (builtin_code == kTfLiteBuiltinArgMax) { in Validate()
1789 Expect(!IsHybridOperator(context, builtin_code, node), in Validate()
[all …]
Dnnapi_delegate_kernel.h266 static TfLiteStatus Map(TfLiteContext* context, int builtin_code, int version,
273 const TfLiteContext* context, int builtin_code, int version,
354 const TfLiteContext* context, int builtin_code, const TfLiteNode* node,
/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/delegates/
Ddelegate_test_util.cc47 reg.builtin_code = tflite::BuiltinOperator_CUSTOM; in AddOpRegistration()
154 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
157 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
171 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
174 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
374 add_reg->builtin_code = kTfLiteBuiltinAdd; in SetUp()
375 deq_reg->builtin_code = kTfLiteBuiltinDequantize; in SetUp()
376 mul_reg->builtin_code = kTfLiteBuiltinMul; in SetUp()
433 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/micro/
Dmicro_mutable_op_resolver.h49 if (registration.builtin_code == op) { in FindOp()
59 if ((registration.builtin_code == BuiltinOperator_CUSTOM) && in FindOp()
107 new_registration->builtin_code = BuiltinOperator_CUSTOM; in AddCustom()
484 registrations_[registrations_len_].builtin_code = op; in AddBuiltin()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport.cc46 auto builtin_code = GetBuiltinCode(opcode); in LoadOperatorsTable() local
47 if (builtin_code != ::tflite::BuiltinOperator_CUSTOM) { in LoadOperatorsTable()
48 operators_table->push_back(EnumNameBuiltinOperator(builtin_code)); in LoadOperatorsTable()
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc676 auto builtin_code = GetBuiltinCode(op_code); in GetOpSignature() local
677 OpSignature op_sig = {builtin_code}; in GetOpSignature()
679 switch (builtin_code) { in GetOpSignature()
860 auto builtin_code = GetBuiltinCode(op_code); in UpdateOpVersion() local
861 if (builtin_code != BuiltinOperator_CUSTOM) { in UpdateOpVersion()
867 << EnumNameBuiltinOperator(builtin_code) << " to version " in UpdateOpVersion()

12345