Searched refs:builtin_code (Results 1 – 10 of 10) sorted by relevance
30 auto builtin_code = GetBuiltinCode(opcode); in GetRegistrationFromOpCode() local33 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()
24 const BuiltinOperator builtin_code);30 BuiltinOperator builtin_code = BuiltinOperator_ADD,36 BuiltinOperator builtin_code = BuiltinOperator_ADD,
50 op_code->builtin_code(), in GetBuiltinCode()58 return std::max(op_code->builtin_code, static_cast<BuiltinOperator>( in GetBuiltinCode()
10124 tflite::BuiltinOperator builtin_code;10128 builtin_code(tflite::BuiltinOperator_ADD) {10149 tflite::BuiltinOperator builtin_code() const {10178 void add_builtin_code(tflite::BuiltinOperator builtin_code) {10179 fbb_.AddElement<int32_t>(OperatorCode::VT_BUILTIN_CODE, static_cast<int32_t>(builtin_code), 0);10198 tflite::BuiltinOperator builtin_code = tflite::BuiltinOperator_ADD) {10200 builder_.add_builtin_code(builtin_code);10212 tflite::BuiltinOperator builtin_code = tflite::BuiltinOperator_ADD) {10219 builtin_code);15018 { auto _e = builtin_code(); _o->builtin_code = _e; }[all …]
31 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in OpNameFromRegistration()34 return EnumNameBuiltinOperator(BuiltinOperator(registration->builtin_code)); in OpNameFromRegistration()71 if (registration->builtin_code == BuiltinOperator_CUSTOM) { in InitSubgraphs()
49 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()552 registrations_[registrations_len_].builtin_code = op; in AddBuiltin()
125 static_cast<BuiltinOperator>(registration->builtin_code); in PrepareNodeAndRegistrationDataFromFlatbuffer()
256 fully_connected_registration.builtin_code = 0; in Register_FULLY_CONNECTED()268 fully_connected_registration.builtin_code = 0; in Register_FULLY_CONNECTED_INT8()
853 int32_t builtin_code; member
157 registration_.builtin_code = static_cast<int32_t>(op); in op_()
Completed in 33 milliseconds