| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_conversion_utils.cc | 42 flatbuffers::Offset<flatbuffers::String> custom_code, int32_t version) { in CreateOperatorCode() argument
52 builder_.add_custom_code(custom_code); in CreateOperatorCode()
59 const char *custom_code, int32_t version) { in CreateOperatorCodeDirect() argument
60 auto custom_code__ = custom_code ? _fbb.CreateString(custom_code) : 0; in CreateOperatorCodeDirect()
|
| D | schema_conversion_utils.h | 31 flatbuffers::Offset<flatbuffers::String> custom_code = 0,
37 const char *custom_code = nullptr, int32_t version = 1);
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export.h | 103 OperatorKey(::tflite::BuiltinOperator type, const std::string& custom_code, in OperatorKey() argument
105 : type_(type), custom_code_(custom_code), version_(version) {} in OperatorKey()
110 const std::string& custom_code() const { return custom_code_; } in custom_code() function
148 std::hash<std::string>()(key.custom_code()), in operator()
|
| D | export_test.cc | 181 opcode->custom_code()->c_str()); in ExportAndSummarizeOperators()
698 EXPECT_EQ(key.custom_code(), ""); in TEST()
720 EXPECT_EQ(key.custom_code(), ""); in TEST()
734 EXPECT_EQ(key.custom_code(), "MyCrazyCustomOp"); in TEST()
749 EXPECT_EQ(key.custom_code(), "BatchMatMul"); in TEST()
761 EXPECT_EQ(key.custom_code(), "FlexBatchMatMul"); in TEST()
778 EXPECT_EQ(key.custom_code(), "FlexMerge"); in TEST()
796 EXPECT_EQ(key.custom_code(), "HashTableV2"); in TEST()
820 EXPECT_EQ(key.custom_code(), "Assert"); in TEST()
836 EXPECT_EQ(key.custom_code(), "FlexAssert"); in TEST()
|
| D | export.cc | 318 flatbuffers::Offset<flatbuffers::String> custom_code = 0; in ExportOperatorCodes() local
319 if (!operator_key.custom_code().empty()) { in ExportOperatorCodes()
320 custom_code = builder->CreateString(operator_key.custom_code()); in ExportOperatorCodes()
324 *builder, operator_key.type(), custom_code, operator_key.version()); in ExportOperatorCodes()
522 custom_ops.insert(key.custom_code()); in Export()
|
| D | import.cc | 50 operators_table->push_back(opcode->custom_code()->c_str()); in LoadOperatorsTable()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | op_resolver.cc | 52 } else if (!opcode->custom_code()) { in GetRegistrationFromOpCode()
58 const char* name = opcode->custom_code()->c_str(); in GetRegistrationFromOpCode()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | custom_op.mlir | 4 …%0 = "tfl.custom"(%arg0, %arg1, %arg2) {custom_code = "Convolution2DTransposeBias", custom_option …
8 // CHECK: "tfl.custom"(%arg0, %arg1, %arg2) {custom_code = "Convolution2DTransposeBias", custom_opt…
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops_no_kernel.cc | 45 !tflite::IsFlexOp(opcode->custom_code()->c_str())) { in AddFlexOpsFromModel()
50 std::string flex_op_name(opcode->custom_code()->c_str()); in AddFlexOpsFromModel()
|
| D | list_flex_ops.cc | 92 !tflite::IsFlexOp(opcode->custom_code()->c_str())) { in AddFlexOpsFromModel()
97 std::string flex_op_name(opcode->custom_code()->c_str()); in AddFlexOpsFromModel()
|
| D | gen_op_registration.cc | 51 opcode->custom_code()->c_str(), std::make_pair(version, version))); in ReadOpsFromModel()
|
| D | verifier.cc | 675 if (IsNullOrEmptyString(opcode->custom_code())) { in VerifyOps()
679 } else if (!resolver.FindOp(opcode->custom_code()->c_str(), in VerifyOps()
682 opcode->custom_code()->c_str(), opcode->version()); in VerifyOps()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights.cc | 85 const std::string custom_code = op_code->custom_code; in GetWeightInputIndices() local
86 const auto& custom_op_info = custom_op_map.find(custom_code); in GetWeightInputIndices()
141 const std::string custom_code = op_code->custom_code; in IsHybridEvaluationOp() local
142 const auto custom_op_info = custom_op_map.find(custom_code); in IsHybridEvaluationOp()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 300 if (opcode.custom_code() != nullptr) { in GetOpName()
301 return opcode.custom_code()->str(); in GetOpName()
367 op_info.is_custom_op = operator_code->custom_code() != nullptr; in BuildLoggingInterpreter()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.h | 69 const std::string &custom_code, const std::vector<uint8_t> &custom_options,
|
| D | flatbuffer_operator.cc | 240 const std::string& custom_code, const std::vector<uint8_t>& custom_options, in CustomOptionsToAttributes() argument
244 builder.getNamedAttr("custom_code", builder.getStringAttr(custom_code))); in CustomOptionsToAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | hashtable_resource.mlir | 7 // CHECK: custom_code: "HashTableV2",
|
| D | type_attr.mlir | 7 // CHECK: custom_code: "SomeOperation",
|
| D | flex_exclusively.mlir | 8 // CHECK-NEXT: custom_code: "FlexAddV2"
|
| D | flex_op_with_f64.mlir | 9 // CHECK-NEXT: custom_code: "FlexAdd",
|
| D | numeric_verify.mlir | 7 // CHECK-NEXT: custom_code: "NumericVerify",
|
| D | flex_op_with_complex128.mlir | 9 // CHECK-NEXT: custom_code: "FlexAdd",
|
| D | custom_op_with_tflite_op.mlir | 14 // CHECK-NEXT: custom_code: "MyCustomOp",
|
| D | flex_op_with_tflite_op.mlir | 13 // CHECK-NEXT: custom_code: "FlexDiv",
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 236 if (!opcode->custom_code()) { in BuildLocalIndexToRegistrationMapping()
241 const auto* op_name = opcode->custom_code()->c_str(); in BuildLocalIndexToRegistrationMapping()
|