tflite_op_resolver.cc 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/optimization_guide/core/tflite_op_resolver.h"
  5. #include "components/optimization_guide/core/optimization_guide_features.h"
  6. #include "components/optimization_guide/machine_learning_tflite_buildflags.h"
  7. #include "third_party/tflite/src/tensorflow/lite/c/common.h"
  8. #include "third_party/tflite/src/tensorflow/lite/kernels/builtin_op_kernels.h"
  9. #include "third_party/tflite/src/tensorflow/lite/schema/schema_generated.h"
  10. #if BUILDFLAG(BUILD_TFLITE_WITH_XNNPACK)
  11. #include "third_party/tflite/src/tensorflow/lite/tflite_with_xnnpack_optional.h"
  12. #endif
  13. namespace optimization_guide {
  14. TFLiteOpResolver::TFLiteOpResolver() {
  15. AddBuiltin(tflite::BuiltinOperator_ABS, tflite::ops::builtin::Register_ABS());
  16. AddBuiltin(tflite::BuiltinOperator_HARD_SWISH,
  17. tflite::ops::builtin::Register_HARD_SWISH());
  18. AddBuiltin(tflite::BuiltinOperator_RELU,
  19. tflite::ops::builtin::Register_RELU(), /* min_version = */ 1,
  20. /* max_version = */ 2);
  21. AddBuiltin(tflite::BuiltinOperator_RELU_N1_TO_1,
  22. tflite::ops::builtin::Register_RELU_N1_TO_1());
  23. AddBuiltin(tflite::BuiltinOperator_RELU6,
  24. tflite::ops::builtin::Register_RELU6(), /* min_version = */ 1,
  25. /* max_version = */ 2);
  26. AddBuiltin(tflite::BuiltinOperator_TANH,
  27. tflite::ops::builtin::Register_TANH(), /* min_version = */ 1,
  28. /* max_version = */ 3);
  29. AddBuiltin(tflite::BuiltinOperator_LOGISTIC,
  30. tflite::ops::builtin::Register_LOGISTIC(),
  31. /* min_version = */ 1,
  32. /* max_version = */ 3);
  33. AddBuiltin(tflite::BuiltinOperator_AVERAGE_POOL_2D,
  34. tflite::ops::builtin::Register_AVERAGE_POOL_2D(),
  35. /* min_version */ 1,
  36. /* max_version */ 3);
  37. AddBuiltin(tflite::BuiltinOperator_MAX_POOL_2D,
  38. tflite::ops::builtin::Register_MAX_POOL_2D(),
  39. /* min_version */ 1,
  40. /* max_version */ 3);
  41. AddBuiltin(tflite::BuiltinOperator_L2_POOL_2D,
  42. tflite::ops::builtin::Register_L2_POOL_2D());
  43. AddBuiltin(tflite::BuiltinOperator_CONV_2D,
  44. tflite::ops::builtin::Register_CONV_2D(),
  45. /* min_version = */ 1,
  46. /* max_version = */ 4);
  47. AddBuiltin(tflite::BuiltinOperator_DEPTHWISE_CONV_2D,
  48. tflite::ops::builtin::Register_DEPTHWISE_CONV_2D(),
  49. /* min_version = */ 1,
  50. /* max_version = */ 5);
  51. AddBuiltin(tflite::BuiltinOperator_SVDF,
  52. tflite::ops::builtin::Register_SVDF(),
  53. /* min_version = */ 1,
  54. /* max_version = */ 3);
  55. AddBuiltin(tflite::BuiltinOperator_RNN, tflite::ops::builtin::Register_RNN(),
  56. /* min_version = */ 1,
  57. /* max_version = */ 2);
  58. AddBuiltin(tflite::BuiltinOperator_BIDIRECTIONAL_SEQUENCE_RNN,
  59. tflite::ops::builtin::Register_BIDIRECTIONAL_SEQUENCE_RNN(),
  60. /* min_version = */ 1,
  61. /* max_version = */ 2);
  62. AddBuiltin(tflite::BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_RNN,
  63. tflite::ops::builtin::Register_UNIDIRECTIONAL_SEQUENCE_RNN(),
  64. /* min_version = */ 1,
  65. /* max_version = */ 2);
  66. AddBuiltin(tflite::BuiltinOperator_EMBEDDING_LOOKUP,
  67. tflite::ops::builtin::Register_EMBEDDING_LOOKUP(),
  68. /* min_version = */ 1,
  69. /* max_version = */ 3);
  70. AddBuiltin(tflite::BuiltinOperator_EMBEDDING_LOOKUP_SPARSE,
  71. tflite::ops::builtin::Register_EMBEDDING_LOOKUP_SPARSE());
  72. AddBuiltin(tflite::BuiltinOperator_FULLY_CONNECTED,
  73. tflite::ops::builtin::Register_FULLY_CONNECTED(),
  74. /* min_version = */ 1,
  75. /* max_version = */ 9);
  76. AddBuiltin(tflite::BuiltinOperator_LSH_PROJECTION,
  77. tflite::ops::builtin::Register_LSH_PROJECTION());
  78. AddBuiltin(tflite::BuiltinOperator_HASHTABLE_LOOKUP,
  79. tflite::ops::builtin::Register_HASHTABLE_LOOKUP());
  80. AddBuiltin(tflite::BuiltinOperator_SOFTMAX,
  81. tflite::ops::builtin::Register_SOFTMAX(),
  82. /* min_version = */ 1,
  83. /* max_version = */ 3);
  84. AddBuiltin(tflite::BuiltinOperator_CONCATENATION,
  85. tflite::ops::builtin::Register_CONCATENATION(),
  86. /* min_version = */ 1,
  87. /* max_version = */ 3);
  88. AddBuiltin(tflite::BuiltinOperator_ADD, tflite::ops::builtin::Register_ADD(),
  89. /* min_version = */ 1,
  90. /* max_version = */ 2);
  91. AddBuiltin(tflite::BuiltinOperator_SPACE_TO_BATCH_ND,
  92. tflite::ops::builtin::Register_SPACE_TO_BATCH_ND(),
  93. /* min_version = */ 1,
  94. /* max_version = */ 3);
  95. AddBuiltin(tflite::BuiltinOperator_BATCH_TO_SPACE_ND,
  96. tflite::ops::builtin::Register_BATCH_TO_SPACE_ND(),
  97. /* min_version = */ 1,
  98. /* max_version = */ 3);
  99. AddBuiltin(tflite::BuiltinOperator_MUL, tflite::ops::builtin::Register_MUL(),
  100. /* min_version = */ 1,
  101. /* max_version = */ 4);
  102. AddBuiltin(tflite::BuiltinOperator_L2_NORMALIZATION,
  103. tflite::ops::builtin::Register_L2_NORMALIZATION(),
  104. /* min_version = */ 1,
  105. /* max_version = */ 2);
  106. AddBuiltin(tflite::BuiltinOperator_LOCAL_RESPONSE_NORMALIZATION,
  107. tflite::ops::builtin::Register_LOCAL_RESPONSE_NORMALIZATION());
  108. AddBuiltin(tflite::BuiltinOperator_LSTM,
  109. tflite::ops::builtin::Register_LSTM(), /* min_version = */ 1,
  110. /* max_version = */ 3);
  111. AddBuiltin(tflite::BuiltinOperator_BIDIRECTIONAL_SEQUENCE_LSTM,
  112. tflite::ops::builtin::Register_BIDIRECTIONAL_SEQUENCE_LSTM(),
  113. /* min_version = */ 1,
  114. /* max_version = */ 3);
  115. AddBuiltin(tflite::BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM,
  116. tflite::ops::builtin::Register_UNIDIRECTIONAL_SEQUENCE_LSTM(),
  117. /* min_version = */ 1,
  118. /* max_version = */ 2);
  119. AddBuiltin(tflite::BuiltinOperator_PAD, tflite::ops::builtin::Register_PAD(),
  120. /* min_version = */ 1,
  121. /* max_version = */ 2);
  122. AddBuiltin(tflite::BuiltinOperator_PADV2,
  123. tflite::ops::builtin::Register_PADV2(), /* min_version = */ 1,
  124. /* max_version = */ 2);
  125. AddBuiltin(tflite::BuiltinOperator_RESHAPE,
  126. tflite::ops::builtin::Register_RESHAPE());
  127. AddBuiltin(tflite::BuiltinOperator_RESIZE_BILINEAR,
  128. tflite::ops::builtin::Register_RESIZE_BILINEAR(),
  129. /* min_version = */ 1,
  130. /* max_version = */ 3);
  131. AddBuiltin(tflite::BuiltinOperator_RESIZE_NEAREST_NEIGHBOR,
  132. tflite::ops::builtin::Register_RESIZE_NEAREST_NEIGHBOR(),
  133. /* min_version = */ 1,
  134. /* max_version = */ 3);
  135. AddBuiltin(tflite::BuiltinOperator_SKIP_GRAM,
  136. tflite::ops::builtin::Register_SKIP_GRAM());
  137. AddBuiltin(tflite::BuiltinOperator_SPACE_TO_DEPTH,
  138. tflite::ops::builtin::Register_SPACE_TO_DEPTH(),
  139. /* min_version = */ 1,
  140. /* max_version = */ 2);
  141. AddBuiltin(tflite::BuiltinOperator_DEPTH_TO_SPACE,
  142. tflite::ops::builtin::Register_DEPTH_TO_SPACE());
  143. AddBuiltin(tflite::BuiltinOperator_GATHER,
  144. tflite::ops::builtin::Register_GATHER(),
  145. /* min_version = */ 1,
  146. /* max_version = */ 3);
  147. AddBuiltin(tflite::BuiltinOperator_TRANSPOSE,
  148. tflite::ops::builtin::Register_TRANSPOSE(),
  149. /* min_version = */ 1,
  150. /* max_version = */ 4);
  151. AddBuiltin(tflite::BuiltinOperator_MEAN,
  152. tflite::ops::builtin::Register_MEAN(),
  153. /* min_version = */ 1,
  154. /* max_version = */ 2);
  155. AddBuiltin(tflite::BuiltinOperator_DIV, tflite::ops::builtin::Register_DIV(),
  156. /* min_version */ 1,
  157. /* max_version */ 2);
  158. AddBuiltin(tflite::BuiltinOperator_SUB, tflite::ops::builtin::Register_SUB(),
  159. /* min_version = */ 1,
  160. /* max_version = */ 3);
  161. AddBuiltin(tflite::BuiltinOperator_SPLIT,
  162. tflite::ops::builtin::Register_SPLIT(),
  163. /* min_version = */ 1,
  164. /* max_version = */ 4);
  165. AddBuiltin(tflite::BuiltinOperator_SPLIT_V,
  166. tflite::ops::builtin::Register_SPLIT_V(),
  167. /* min_version = */ 1,
  168. /* max_version = */ 2);
  169. AddBuiltin(tflite::BuiltinOperator_SQUEEZE,
  170. tflite::ops::builtin::Register_SQUEEZE());
  171. AddBuiltin(tflite::BuiltinOperator_STRIDED_SLICE,
  172. tflite::ops::builtin::Register_STRIDED_SLICE(),
  173. /* min_version = */ 1,
  174. /* max_version = */ 4);
  175. AddBuiltin(tflite::BuiltinOperator_EXP, tflite::ops::builtin::Register_EXP());
  176. AddBuiltin(tflite::BuiltinOperator_TOPK_V2,
  177. tflite::ops::builtin::Register_TOPK_V2(),
  178. /* min_version = */ 1,
  179. /* max_version = */ 2);
  180. AddBuiltin(tflite::BuiltinOperator_LOG, tflite::ops::builtin::Register_LOG());
  181. AddBuiltin(tflite::BuiltinOperator_LOG_SOFTMAX,
  182. tflite::ops::builtin::Register_LOG_SOFTMAX(),
  183. /* min_version = */ 1,
  184. /* max_version = */ 2);
  185. AddBuiltin(tflite::BuiltinOperator_CAST,
  186. tflite::ops::builtin::Register_CAST());
  187. AddBuiltin(tflite::BuiltinOperator_DEQUANTIZE,
  188. tflite::ops::builtin::Register_DEQUANTIZE(),
  189. /* min_version = */ 1,
  190. /* max_version = */ 4);
  191. AddBuiltin(tflite::BuiltinOperator_PRELU,
  192. tflite::ops::builtin::Register_PRELU());
  193. AddBuiltin(tflite::BuiltinOperator_MAXIMUM,
  194. tflite::ops::builtin::Register_MAXIMUM(),
  195. /* min_version = */ 1,
  196. /* max_version = */ 4);
  197. AddBuiltin(tflite::BuiltinOperator_MINIMUM,
  198. tflite::ops::builtin::Register_MINIMUM(),
  199. /* min_version = */ 1,
  200. /* max_version = */ 4);
  201. AddBuiltin(tflite::BuiltinOperator_ARG_MAX,
  202. tflite::ops::builtin::Register_ARG_MAX(),
  203. /* min_version = */ 1,
  204. /* max_version = */ 2);
  205. AddBuiltin(tflite::BuiltinOperator_ARG_MIN,
  206. tflite::ops::builtin::Register_ARG_MIN(),
  207. /* min_version = */ 1,
  208. /* max_version = */ 2);
  209. AddBuiltin(tflite::BuiltinOperator_GREATER,
  210. tflite::ops::builtin::Register_GREATER(),
  211. /* min_version = */ 1,
  212. /* max_version = */ 2);
  213. AddBuiltin(tflite::BuiltinOperator_GREATER_EQUAL,
  214. tflite::ops::builtin::Register_GREATER_EQUAL(),
  215. /* min_version = */ 1,
  216. /* max_version = */ 2);
  217. AddBuiltin(tflite::BuiltinOperator_LESS,
  218. tflite::ops::builtin::Register_LESS(),
  219. /* min_version = */ 1,
  220. /* max_version = */ 2);
  221. AddBuiltin(tflite::BuiltinOperator_LESS_EQUAL,
  222. tflite::ops::builtin::Register_LESS_EQUAL(),
  223. /* min_version = */ 1,
  224. /* max_version = */ 2);
  225. AddBuiltin(tflite::BuiltinOperator_FLOOR,
  226. tflite::ops::builtin::Register_FLOOR());
  227. AddBuiltin(tflite::BuiltinOperator_CEIL,
  228. tflite::ops::builtin::Register_CEIL());
  229. AddBuiltin(tflite::BuiltinOperator_ROUND,
  230. tflite::ops::builtin::Register_ROUND());
  231. AddBuiltin(tflite::BuiltinOperator_NEG, tflite::ops::builtin::Register_NEG());
  232. AddBuiltin(tflite::BuiltinOperator_SELECT,
  233. tflite::ops::builtin::Register_SELECT(),
  234. /* min_version = */ 1,
  235. /* max_version = */ 2);
  236. AddBuiltin(tflite::BuiltinOperator_SELECT_V2,
  237. tflite::ops::builtin::Register_SELECT_V2());
  238. AddBuiltin(tflite::BuiltinOperator_SLICE,
  239. tflite::ops::builtin::Register_SLICE(),
  240. /* min_version = */ 1,
  241. /* max_version = */ 3);
  242. AddBuiltin(tflite::BuiltinOperator_SIN, tflite::ops::builtin::Register_SIN());
  243. AddBuiltin(tflite::BuiltinOperator_COS, tflite::ops::builtin::Register_COS());
  244. AddBuiltin(tflite::BuiltinOperator_TRANSPOSE_CONV,
  245. tflite::ops::builtin::Register_TRANSPOSE_CONV(),
  246. /* min_version = */ 1,
  247. /* max_version = */ 3);
  248. AddBuiltin(tflite::BuiltinOperator_TILE,
  249. tflite::ops::builtin::Register_TILE(),
  250. /* min_version = */ 1,
  251. /* max_version = */ 2);
  252. AddBuiltin(tflite::BuiltinOperator_SUM, tflite::ops::builtin::Register_SUM(),
  253. /* min_version = */ 1,
  254. /* max_version = */ 2);
  255. AddBuiltin(tflite::BuiltinOperator_REDUCE_PROD,
  256. tflite::ops::builtin::Register_REDUCE_PROD());
  257. AddBuiltin(tflite::BuiltinOperator_REDUCE_MAX,
  258. tflite::ops::builtin::Register_REDUCE_MAX(),
  259. /* min_version = */ 1,
  260. /* max_version = */ 2);
  261. AddBuiltin(tflite::BuiltinOperator_REDUCE_MIN,
  262. tflite::ops::builtin::Register_REDUCE_MIN(),
  263. /* min_version = */ 1,
  264. /* max_version = */ 2);
  265. AddBuiltin(tflite::BuiltinOperator_REDUCE_ANY,
  266. tflite::ops::builtin::Register_REDUCE_ANY());
  267. AddBuiltin(tflite::BuiltinOperator_EXPAND_DIMS,
  268. tflite::ops::builtin::Register_EXPAND_DIMS());
  269. AddBuiltin(tflite::BuiltinOperator_SPARSE_TO_DENSE,
  270. tflite::ops::builtin::Register_SPARSE_TO_DENSE(),
  271. /* min_version = */ 1,
  272. /* max_version = */ 3);
  273. AddBuiltin(tflite::BuiltinOperator_EQUAL,
  274. tflite::ops::builtin::Register_EQUAL(),
  275. /* min_version = */ 1,
  276. /* max_version = */ 3);
  277. AddBuiltin(tflite::BuiltinOperator_NOT_EQUAL,
  278. tflite::ops::builtin::Register_NOT_EQUAL(),
  279. /* min_version = */ 1,
  280. /* max_version = */ 3);
  281. AddBuiltin(tflite::BuiltinOperator_SQRT,
  282. tflite::ops::builtin::Register_SQRT());
  283. AddBuiltin(tflite::BuiltinOperator_RSQRT,
  284. tflite::ops::builtin::Register_RSQRT());
  285. AddBuiltin(tflite::BuiltinOperator_SHAPE,
  286. tflite::ops::builtin::Register_SHAPE());
  287. AddBuiltin(tflite::BuiltinOperator_RANK,
  288. tflite::ops::builtin::Register_RANK());
  289. AddBuiltin(tflite::BuiltinOperator_POW, tflite::ops::builtin::Register_POW());
  290. AddBuiltin(tflite::BuiltinOperator_FAKE_QUANT,
  291. tflite::ops::builtin::Register_FAKE_QUANT(), 1, 2);
  292. AddBuiltin(tflite::BuiltinOperator_PACK,
  293. tflite::ops::builtin::Register_PACK(),
  294. /* min_version = */ 1,
  295. /* max_version = */ 3);
  296. AddBuiltin(tflite::BuiltinOperator_ONE_HOT,
  297. tflite::ops::builtin::Register_ONE_HOT());
  298. AddBuiltin(tflite::BuiltinOperator_LOGICAL_OR,
  299. tflite::ops::builtin::Register_LOGICAL_OR());
  300. AddBuiltin(tflite::BuiltinOperator_LOGICAL_AND,
  301. tflite::ops::builtin::Register_LOGICAL_AND());
  302. AddBuiltin(tflite::BuiltinOperator_LOGICAL_NOT,
  303. tflite::ops::builtin::Register_LOGICAL_NOT());
  304. AddBuiltin(tflite::BuiltinOperator_UNPACK,
  305. tflite::ops::builtin::Register_UNPACK(),
  306. /* min_version = */ 1,
  307. /* max_version = */ 4);
  308. AddBuiltin(tflite::BuiltinOperator_FLOOR_DIV,
  309. tflite::ops::builtin::Register_FLOOR_DIV(),
  310. /* min_version = */ 1,
  311. /* max_version = */ 2);
  312. AddBuiltin(tflite::BuiltinOperator_SQUARE,
  313. tflite::ops::builtin::Register_SQUARE());
  314. AddBuiltin(tflite::BuiltinOperator_ZEROS_LIKE,
  315. tflite::ops::builtin::Register_ZEROS_LIKE());
  316. AddBuiltin(tflite::BuiltinOperator_FLOOR_MOD,
  317. tflite::ops::builtin::Register_FLOOR_MOD());
  318. AddBuiltin(tflite::BuiltinOperator_RANGE,
  319. tflite::ops::builtin::Register_RANGE());
  320. AddBuiltin(tflite::BuiltinOperator_LEAKY_RELU,
  321. tflite::ops::builtin::Register_LEAKY_RELU(),
  322. /* min_version = */ 1,
  323. /* max_version = */ 2);
  324. AddBuiltin(tflite::BuiltinOperator_SQUARED_DIFFERENCE,
  325. tflite::ops::builtin::Register_SQUARED_DIFFERENCE());
  326. AddBuiltin(tflite::BuiltinOperator_FILL,
  327. tflite::ops::builtin::Register_FILL(),
  328. /* min_version = */ 1,
  329. /* max_version = */ 2);
  330. AddBuiltin(tflite::BuiltinOperator_MIRROR_PAD,
  331. tflite::ops::builtin::Register_MIRROR_PAD(),
  332. /* min_version = */ 1,
  333. /* max_version = */ 2);
  334. AddBuiltin(tflite::BuiltinOperator_UNIQUE,
  335. tflite::ops::builtin::Register_UNIQUE());
  336. AddBuiltin(tflite::BuiltinOperator_REVERSE_V2,
  337. tflite::ops::builtin::Register_REVERSE_V2(),
  338. /* min_version = */ 1,
  339. /* max_version = */ 2);
  340. AddBuiltin(tflite::BuiltinOperator_ADD_N,
  341. tflite::ops::builtin::Register_ADD_N());
  342. AddBuiltin(tflite::BuiltinOperator_GATHER_ND,
  343. tflite::ops::builtin::Register_GATHER_ND(),
  344. /* min_version = */ 1,
  345. /* max_version = */ 2);
  346. AddBuiltin(tflite::BuiltinOperator_WHERE,
  347. tflite::ops::builtin::Register_WHERE());
  348. AddBuiltin(tflite::BuiltinOperator_ELU, tflite::ops::builtin::Register_ELU());
  349. AddBuiltin(tflite::BuiltinOperator_REVERSE_SEQUENCE,
  350. tflite::ops::builtin::Register_REVERSE_SEQUENCE());
  351. AddBuiltin(tflite::BuiltinOperator_MATRIX_DIAG,
  352. tflite::ops::builtin::Register_MATRIX_DIAG());
  353. AddBuiltin(tflite::BuiltinOperator_QUANTIZE,
  354. tflite::ops::builtin::Register_QUANTIZE(),
  355. /* min_version = */ 1,
  356. /* max_version = */ 2);
  357. AddBuiltin(tflite::BuiltinOperator_MATRIX_SET_DIAG,
  358. tflite::ops::builtin::Register_MATRIX_SET_DIAG());
  359. AddBuiltin(tflite::BuiltinOperator_IF, tflite::ops::builtin::Register_IF());
  360. AddBuiltin(tflite::BuiltinOperator_WHILE,
  361. tflite::ops::builtin::Register_WHILE());
  362. AddBuiltin(tflite::BuiltinOperator_NON_MAX_SUPPRESSION_V4,
  363. tflite::ops::builtin::Register_NON_MAX_SUPPRESSION_V4());
  364. AddBuiltin(tflite::BuiltinOperator_NON_MAX_SUPPRESSION_V5,
  365. tflite::ops::builtin::Register_NON_MAX_SUPPRESSION_V5());
  366. AddBuiltin(tflite::BuiltinOperator_SCATTER_ND,
  367. tflite::ops::builtin::Register_SCATTER_ND());
  368. AddBuiltin(tflite::BuiltinOperator_DENSIFY,
  369. tflite::ops::builtin::Register_DENSIFY());
  370. AddBuiltin(tflite::BuiltinOperator_SEGMENT_SUM,
  371. tflite::ops::builtin::Register_SEGMENT_SUM());
  372. AddBuiltin(tflite::BuiltinOperator_BATCH_MATMUL,
  373. tflite::ops::builtin::Register_BATCH_MATMUL(),
  374. /* min_version = */ 1,
  375. /* max_version = */ 4);
  376. AddBuiltin(tflite::BuiltinOperator_GELU,
  377. tflite::ops::builtin::Register_GELU(),
  378. /* min_version = */ 1,
  379. /* max_version = */ 2);
  380. #if BUILDFLAG(BUILD_TFLITE_WITH_XNNPACK)
  381. if (features::TFLiteXNNPACKDelegateEnabled()) {
  382. delegate_creators_.push_back([](int num_threads) {
  383. return tflite::MaybeCreateXNNPACKDelegate(num_threads);
  384. });
  385. }
  386. #endif
  387. }
  388. } // namespace optimization_guide