tflite-metadata.json 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197
  1. [
  2. {
  3. "name": "Add",
  4. "attributes": [
  5. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  6. { "name": "pot_scale_int16", "type": "boolean", "default": true }
  7. ],
  8. "inputs": [
  9. { "name": "A", "type": "T" },
  10. { "name": "B", "type": "T" }
  11. ],
  12. "outputs": [
  13. { "name": "C", "type": "T" }
  14. ]
  15. },
  16. {
  17. "name": "ArgMax",
  18. "attributes": [
  19. { "name": "output_type", "type": "TensorType", "default": "FLOAT32" }
  20. ],
  21. "inputs": [
  22. { "name": "input", "type": "T" },
  23. { "name": "dimension", "type": "T" }
  24. ],
  25. "outputs": [
  26. { "name": "output", "type": "T" }
  27. ]
  28. },
  29. {
  30. "name": "ArgMin",
  31. "attributes": [
  32. { "name": "output_type", "type": "TensorType", "default": "FLOAT32" }
  33. ]
  34. },
  35. {
  36. "name": "AveragePool2D",
  37. "category": "Pool",
  38. "attributes": [
  39. { "name": "padding", "type": "Padding", "default": "SAME" },
  40. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  41. { "name": "stride_w", "type": "int32" },
  42. { "name": "stride_h", "type": "int32" },
  43. { "name": "filter_width", "type": "int32" },
  44. { "name": "filter_height", "type": "int32" }
  45. ],
  46. "inputs": [
  47. { "name": "input", "type": "T" }
  48. ],
  49. "outputs": [
  50. { "name": "output", "type": "T" }
  51. ]
  52. },
  53. {
  54. "name": "BatchMatMul",
  55. "attributes": [
  56. { "name": "adj_x", "type": "boolean", "default": false },
  57. { "name": "adj_y", "type": "boolean", "default": false },
  58. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
  59. ]
  60. },
  61. {
  62. "name": "BidirectionalSequenceLSTM",
  63. "category": "Layer",
  64. "attributes": [
  65. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  66. { "name": "cell_clip", "type": "float32", "default": 0 },
  67. { "name": "proj_clip", "type": "float32", "default": 0 },
  68. { "name": "merge_outputs", "type": "boolean", "default": false },
  69. { "name": "time_major", "type": "boolean", "default": true },
  70. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
  71. ]
  72. },
  73. {
  74. "name": "BidirectionalSequenceRNN",
  75. "attributes": [
  76. { "name": "time_major", "type": "boolean", "default": false },
  77. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  78. { "name": "merge_outputs", "type": "boolean", "default": false },
  79. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
  80. ]
  81. },
  82. {
  83. "name": "Bucketize",
  84. "attributes": [
  85. { "name": "boundaries", "type": "float32[]", "default": 0 }
  86. ]
  87. },
  88. {
  89. "name": "Call",
  90. "attributes": [
  91. { "name": "subgraph", "type": "uint32", "default": 0 }
  92. ]
  93. },
  94. {
  95. "name": "CallOnce",
  96. "attributes": [
  97. { "name": "init_subgraph_index", "type": "int32", "default": 0 }
  98. ]
  99. },
  100. {
  101. "name": "Cast",
  102. "attributes": [
  103. { "name": "in_data_type", "type": "TensorType", "default": "FLOAT32" },
  104. { "name": "out_data_type", "type": "TensorType", "default": "FLOAT32" }
  105. ],
  106. "inputs": [
  107. { "name": "input" }
  108. ],
  109. "outputs": [
  110. { "name": "output" }
  111. ]
  112. },
  113. {
  114. "name": "ConcatEmbeddings",
  115. "attributes": [
  116. { "name": "num_channels", "type": "int32", "default": 0 },
  117. { "name": "num_columns_per_channel", "type": "int32[]", "default": 0 },
  118. { "name": "embedding_dim_per_channel", "type": "int32[]", "default": 0 }
  119. ]
  120. },
  121. {
  122. "name": "Concatenation",
  123. "category": "Tensor",
  124. "attributes": [
  125. { "name": "axis", "type": "int32", "default": 0 },
  126. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
  127. ],
  128. "inputs": [
  129. { "name": "inputs", "list": true }
  130. ],
  131. "outputs": [
  132. { "name": "output" }
  133. ]
  134. },
  135. {
  136. "name": "Conv2D",
  137. "category": "Layer",
  138. "attributes": [
  139. { "name": "padding", "type": "Padding", "default": "SAME", "description": "`SAME`|`VALID`" },
  140. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE", "description": "`NONE`|`RELU`|`RELU6`" },
  141. { "name": "stride_w", "type": "int32", "default": 0, "description": "stride of the filter window" },
  142. { "name": "stride_h", "type": "int32", "default": 0, "description": "stride of the filter window" },
  143. { "name": "dilation_w_factor", "type": "int32", "default": 1 },
  144. { "name": "dilation_h_factor", "type": "int32", "default": 1 },
  145. { "name": "quantized_bias_type", "type": "TensorType", "default": "FLOAT32" }
  146. ],
  147. "inputs": [
  148. { "name": "input", "description": "4D tensor" },
  149. { "name": "filter" },
  150. { "name": "bias", "description": "(optional)" }
  151. ],
  152. "outputs": [
  153. { "name": "output", "description": "result of 2D convolution of the input tensor" }
  154. ]
  155. },
  156. {
  157. "name": "Cumsum",
  158. "attributes": [
  159. { "name": "exclusive", "type": "boolean", "default": false },
  160. { "name": "reverse", "type": "boolean", "default": false }
  161. ]
  162. },
  163. {
  164. "name": "Densify",
  165. "inputs": [
  166. { "name": "input" }
  167. ],
  168. "outputs": [
  169. { "name": "output" }
  170. ]
  171. },
  172. {
  173. "name": "DepthToSpace",
  174. "attributes": [
  175. { "name": "block_size", "type": "int32", "default": 0 }
  176. ]
  177. },
  178. {
  179. "name": "DepthwiseConv2D",
  180. "category": "Layer",
  181. "attributes": [
  182. { "name": "padding", "type": "Padding", "default": "SAME" },
  183. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  184. { "name": "stride_w", "type": "int32", "default": 0 },
  185. { "name": "stride_h", "type": "int32", "default": 0 },
  186. { "name": "depth_multiplier", "type": "int32", "default": 0 },
  187. { "name": "dilation_w_factor", "type": "int32", "default": 1 },
  188. { "name": "dilation_h_factor", "type": "int32", "default": 1 }
  189. ],
  190. "inputs": [
  191. { "name": "input", "type": "T" },
  192. { "name": "weights", "type": "T" },
  193. { "name": "bias", "type": "T" }
  194. ],
  195. "outputs": [
  196. { "name": "output", "type": "T" }
  197. ]
  198. },
  199. {
  200. "name": "Dequantize",
  201. "inputs": [
  202. { "name": "input" }
  203. ],
  204. "outputs": [
  205. { "name": "output" }
  206. ]
  207. },
  208. {
  209. "name": "Div",
  210. "attributes": [
  211. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
  212. ],
  213. "inputs": [
  214. { "name": "A", "type": "T" },
  215. { "name": "B", "type": "T" }
  216. ],
  217. "outputs": [
  218. { "name": "C", "type": "T" }
  219. ]
  220. },
  221. {
  222. "name": "EmbeddingLookupSparse",
  223. "attributes": [
  224. { "name": "combiner", "type": "CombinerType", "default": "SUM" }
  225. ]
  226. },
  227. {
  228. "name": "Exp",
  229. "inputs": [
  230. { "name": "input" }
  231. ],
  232. "outputs": [
  233. { "name": "output" }
  234. ]
  235. },
  236. {
  237. "name": "ExpandDims",
  238. "inputs": [
  239. { "name": "input" },
  240. { "name": "axis_param" }
  241. ],
  242. "outputs": [
  243. { "name": "output" }
  244. ]
  245. },
  246. {
  247. "name": "ExtractFeatures",
  248. "inputs": [
  249. { "name": "ngrams" }
  250. ],
  251. "outputs": [
  252. { "name": "features" },
  253. { "name": "weights" }
  254. ]
  255. },
  256. {
  257. "name": "FakeQuant",
  258. "attributes": [
  259. { "name": "min", "type": "float32", "default": 0 },
  260. { "name": "max", "type": "float32", "default": 0 },
  261. { "name": "num_bits", "type": "int32", "default": 0 },
  262. { "name": "narrow_range", "type": "boolean", "default": false }
  263. ]
  264. },
  265. {
  266. "name": "Fill",
  267. "inputs": [
  268. { "name": "dims", "type": "T" },
  269. { "name": "value", "type": "T" }
  270. ],
  271. "outputs": [
  272. { "name": "output", "type": "T" }
  273. ]
  274. },
  275. {
  276. "name": "FullyConnected",
  277. "category": "Layer",
  278. "attributes": [
  279. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  280. { "name": "weights_format", "type": "FullyConnectedOptionsWeightsFormat", "default": "DEFAULT" },
  281. { "name": "keep_num_dims", "type": "boolean", "default": false },
  282. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false },
  283. { "name": "quantized_bias_type", "type": "TensorType", "default": "FLOAT32" }
  284. ],
  285. "inputs": [
  286. { "name": "input", "type": "T" },
  287. { "name": "weights", "type": "T" },
  288. { "name": "bias", "type": "T" }
  289. ],
  290. "outputs": [
  291. { "name": "output", "type": "T" }
  292. ]
  293. },
  294. {
  295. "name": "Gather",
  296. "category": "Transform",
  297. "attributes": [
  298. { "name": "axis", "default": 0, "type": "int32" },
  299. { "name": "batch_dims", "type": "int32", "default": 0 }
  300. ],
  301. "inputs": [
  302. { "name": "input" },
  303. { "name": "positions" }
  304. ],
  305. "outputs": [
  306. { "name": "output" }
  307. ]
  308. },
  309. {
  310. "name": "Gelu",
  311. "category": "Activation",
  312. "attributes": [
  313. { "name": "approximate", "type": "boolean", "default": false }
  314. ]
  315. },
  316. {
  317. "name": "HardSwish",
  318. "category": "Activation",
  319. "inputs": [
  320. { "name": "input" }
  321. ],
  322. "outputs": [
  323. { "name": "output" }
  324. ]
  325. },
  326. {
  327. "name": "Hashtable",
  328. "attributes": [
  329. { "name": "table_id", "type": "int32", "default": 0 },
  330. { "name": "key_dtype", "type": "TensorType", "default": "FLOAT32" },
  331. { "name": "value_dtype", "type": "TensorType", "default": "FLOAT32" }
  332. ]
  333. },
  334. {
  335. "name": "HashtableLookup",
  336. "inputs": [
  337. { "name": "key" },
  338. { "name": "keys" },
  339. { "name": "values" }
  340. ],
  341. "outputs": [
  342. { "name": "value" },
  343. { "name": "hits" }
  344. ]
  345. },
  346. {
  347. "name": "If",
  348. "attributes": [
  349. { "name": "then_subgraph_index", "type": "int32", "default": 0 },
  350. { "name": "else_subgraph_index", "type": "int32", "default": 0 }
  351. ]
  352. },
  353. {
  354. "name": "InstanceNorm",
  355. "attributes": [
  356. { "name": "epsilon", "type": "float32", "default": 0 },
  357. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
  358. ]
  359. },
  360. {
  361. "name": "LeakyRelu",
  362. "attributes": [
  363. { "name": "alpha", "type": "float32", "default": 0 }
  364. ]
  365. },
  366. {
  367. "name": "LocalResponseNormalization",
  368. "category": "Normalization",
  369. "attributes": [
  370. { "name": "radius", "type": "int32", "default": 0 },
  371. { "name": "bias", "type": "float32", "default": 0 },
  372. { "name": "alpha", "type": "float32", "default": 0 },
  373. { "name": "beta", "type": "float32", "default": 0 }
  374. ],
  375. "inputs": [
  376. { "name": "input" }
  377. ],
  378. "outputs": [
  379. { "name": "output" }
  380. ]
  381. },
  382. {
  383. "name": "LogicalOr",
  384. "inputs": [
  385. { "name": "A", "type": "T" },
  386. { "name": "B", "type": "T" }
  387. ],
  388. "outputs": [
  389. { "name": "C", "type": "T" }
  390. ]
  391. },
  392. {
  393. "name": "Logistic",
  394. "inputs": [
  395. { "name": "input" }
  396. ],
  397. "outputs": [
  398. { "name": "output" }
  399. ]
  400. },
  401. {
  402. "name": "LogSoftmax",
  403. "category": "Activation",
  404. "inputs": [
  405. { "name": "input", "type": "T" }
  406. ],
  407. "outputs": [
  408. { "name": "output", "type": "T" }
  409. ]
  410. },
  411. {
  412. "name": "LSHProjection",
  413. "attributes": [
  414. { "name": "type", "type": "LSHProjectionType", "default": "UNKNOWN" }
  415. ],
  416. "inputs": [
  417. { "name": "hash" },
  418. { "name": "input" },
  419. { "name": "weight" }
  420. ],
  421. "outputs": [
  422. { "name": "output" }
  423. ]
  424. },
  425. {
  426. "name": "LSTM",
  427. "category": "Layer",
  428. "attributes": [
  429. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  430. { "name": "cell_clip", "type": "float32", "default": 0 },
  431. { "name": "proj_clip", "type": "float32", "default": 0 },
  432. { "name": "kernel_type", "type": "LSTMKernelType", "default": "FULL" },
  433. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
  434. ],
  435. "inputs": [
  436. { "name": "input", "type": "T", "description": "Input tensor." },
  437. { "name": "input_input_weights", "type": "T", "option": "optional", "description": "Input to input weights tensor.", "visible": false },
  438. { "name": "input_forget_weights", "type": "T", "description": "Input to forget weights tensor.", "visible": false },
  439. { "name": "input_cell_weights", "type": "T", "description": "Input to cell weights tensor.", "visible": false },
  440. { "name": "input_output_weights", "type": "T", "description": "Input to output weights tensor.", "visible": false },
  441. { "name": "recurrent_input_weights", "type": "T", "option": "optional", "description": "Recurrent to input weights tensor.", "visible": false },
  442. { "name": "recurrent_forget_weights", "type": "T", "description": "Recurrent to forget weights tensor.", "visible": false },
  443. { "name": "recurrent_cell_weights", "type": "T", "description": "Recurrent to cell weights tensor.", "visible": false },
  444. { "name": "recurrent_output_weights", "type": "T", "description": "Recurrent to output weights tensor.", "visible": false },
  445. { "name": "cell_input_weights", "type": "T", "option": "optional", "description": "Cell to input weights tensor.", "visible": false },
  446. { "name": "cell_forget_weights", "type": "T", "option": "optional", "description": "Cell to forget weights tensor.", "visible": false },
  447. { "name": "cell_output_weights", "type": "T", "option": "optional", "description": "Cell to output weights tensor.", "visible": false },
  448. { "name": "input_gate_bias", "type": "T", "option": "optional", "description": "Input gate bias tensor.", "visible": false },
  449. { "name": "forget_gate_bias", "type": "T", "description": "Forget gate bias tensor.", "visible": false },
  450. { "name": "cell_gate_bias", "type": "T", "description": "Cell gate bias tensor.", "visible": false },
  451. { "name": "output_gate_bias", "type": "T", "description": "Output gate bias tensor.", "visible": false },
  452. { "name": "projection_weights", "type": "T", "option": "optional", "description": "Projection weights tensor.", "visible": false },
  453. { "name": "projection_bias", "type": "T", "option": "optional", "description": "Projection bias tensor.", "visible": false }
  454. ],
  455. "outputs": [
  456. { "name": "scratch", "type": "T" },
  457. { "name": "output_state", "type": "T" },
  458. { "name": "cell_state", "type": "T" },
  459. { "name": "output", "type": "T" }
  460. ]
  461. },
  462. {
  463. "name": "Maximum",
  464. "inputs": [
  465. { "name": "input1" },
  466. { "name": "input2" }
  467. ],
  468. "outputs": [
  469. { "name": "output" }
  470. ]
  471. },
  472. {
  473. "name": "MaxPool2D",
  474. "category": "Pool",
  475. "attributes": [
  476. { "name": "padding", "type": "Padding", "default": "SAME" },
  477. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  478. { "name": "stride_w", "type": "int32" },
  479. { "name": "stride_h", "type": "int32" },
  480. { "name": "filter_width", "type": "int32" },
  481. { "name": "filter_height", "type": "int32" }
  482. ],
  483. "inputs": [
  484. { "name": "input", "type": "T" }
  485. ],
  486. "outputs": [
  487. { "name": "output", "type": "T" }
  488. ]
  489. },
  490. {
  491. "name": "Mean",
  492. "attributes": [
  493. { "name": "keep_dims", "type": "boolean" }
  494. ],
  495. "inputs": [
  496. { "name": "input", "type": "T" },
  497. { "name": "axis", "type": "T" }
  498. ],
  499. "outputs": [
  500. { "name": "output", "type": "T" }
  501. ]
  502. },
  503. {
  504. "name": "Minimum",
  505. "inputs": [
  506. { "name": "input1" },
  507. { "name": "input2" }
  508. ],
  509. "outputs": [
  510. { "name": "output" }
  511. ]
  512. },
  513. {
  514. "name": "MirrorPad",
  515. "attributes": [
  516. { "name": "mode", "type": "MirrorPadMode", "default": "REFLECT" }
  517. ]
  518. },
  519. {
  520. "name": "Mul",
  521. "attributes": [
  522. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
  523. ],
  524. "inputs": [
  525. { "name": "A", "type": "T" },
  526. { "name": "B", "type": "T" }
  527. ],
  528. "outputs": [
  529. { "name": "C", "type": "T" }
  530. ]
  531. },
  532. {
  533. "name": "Normalize",
  534. "category": "Normalization",
  535. "inputs": [
  536. { "name": "input" }
  537. ],
  538. "outputs": [
  539. { "name": "output" }
  540. ]
  541. },
  542. {
  543. "name": "NotEqual",
  544. "inputs": [
  545. { "name": "A", "type": "T" },
  546. { "name": "B", "type": "T" }
  547. ],
  548. "outputs": [
  549. { "name": "C" }
  550. ]
  551. },
  552. {
  553. "name": "OneHot",
  554. "attributes": [
  555. { "name": "axis", "type": "int32", "default": 0 }
  556. ]
  557. },
  558. {
  559. "name": "Pack",
  560. "attributes": [
  561. { "name": "values_count", "type": "int32", "default": 0 },
  562. { "name": "axis", "type": "int32", "default": 0 }
  563. ]
  564. },
  565. {
  566. "name": "Pad",
  567. "category": "Tensor",
  568. "inputs": [
  569. { "name": "input" },
  570. { "name": "paddings" }
  571. ],
  572. "outputs": [
  573. { "name": "output" }
  574. ]
  575. },
  576. {
  577. "name": "Predict",
  578. "inputs": [
  579. { "name": "hashes" },
  580. { "name": "keys" },
  581. { "name": "labels" },
  582. { "name": "weights" }
  583. ],
  584. "outputs": [
  585. { "name": "label" },
  586. { "name": "weight" }
  587. ]
  588. },
  589. {
  590. "name": "Prelu",
  591. "category": "Activation",
  592. "inputs": [
  593. { "name": "input", "type": "T" },
  594. { "name": "slope", "type": "T" }
  595. ],
  596. "outputs": [
  597. { "name": "output", "type": "T" }
  598. ]
  599. },
  600. {
  601. "name": "Quantize",
  602. "inputs": [
  603. { "name": "input" }
  604. ],
  605. "outputs": [
  606. { "name": "output" }
  607. ]
  608. },
  609. {
  610. "name": "Range",
  611. "attributes": [
  612. { "name": "keep_dims", "type": "boolean" }
  613. ],
  614. "inputs": [
  615. { "name": "start", "type": "T" },
  616. { "name": "limit", "type": "T" },
  617. { "name": "delta", "type": "T" }
  618. ],
  619. "outputs": [
  620. { "name": "output", "type": "T" }
  621. ]
  622. },
  623. {
  624. "name": "ReduceMax",
  625. "attributes": [
  626. { "name": "keep_dims", "type": "boolean" }
  627. ],
  628. "inputs": [
  629. { "name": "input", "type": "T" },
  630. { "name": "axis", "type": "T" }
  631. ],
  632. "outputs": [
  633. { "name": "output", "type": "T" }
  634. ]
  635. },
  636. {
  637. "name": "ReduceMin",
  638. "attributes": [
  639. { "name": "keep_dims", "type": "boolean" }
  640. ],
  641. "inputs": [
  642. { "name": "input", "type": "T" },
  643. { "name": "axis", "type": "T" }
  644. ],
  645. "outputs": [
  646. { "name": "output", "type": "T" }
  647. ]
  648. },
  649. {
  650. "name": "ReduceWindow",
  651. "attributes": [
  652. { "name": "reduce_function", "type": "ReduceWindowFunction", "default": "UNSUPPORTED" }
  653. ]
  654. },
  655. {
  656. "name": "Relu",
  657. "category": "Activation",
  658. "inputs": [
  659. { "name": "input", "type": "T" }
  660. ],
  661. "outputs": [
  662. { "name": "output", "type": "T" }
  663. ]
  664. },
  665. {
  666. "name": "Relu6",
  667. "category": "Activation",
  668. "inputs": [
  669. { "name": "input", "type": "T" }
  670. ],
  671. "outputs": [
  672. { "name": "output", "type": "T" }
  673. ]
  674. },
  675. {
  676. "name": "Reshape",
  677. "category": "Shape",
  678. "attributes": [
  679. { "name": "new_shape", "type": "int32[]", "default": 0 }
  680. ],
  681. "inputs": [
  682. { "name": "data", "type": "T" },
  683. { "name": "shape", "type": "T" }
  684. ],
  685. "outputs": [
  686. { "name": "reshaped", "type": "T" }
  687. ]
  688. },
  689. {
  690. "name": "ResizeBilinear",
  691. "attributes": [
  692. { "name": "align_corners", "default": false, "type": "boolean" },
  693. { "name": "new_height", "type": "int32", "default": 0 },
  694. { "name": "new_width", "type": "int32", "default": 0 },
  695. { "name": "half_pixel_centers", "type": "boolean", "default": false }
  696. ],
  697. "inputs": [
  698. { "name": "input" },
  699. { "name": "size" }
  700. ],
  701. "outputs": [
  702. { "name": "output" }
  703. ]
  704. },
  705. {
  706. "name": "ResizeNearestNeighbor",
  707. "attributes": [
  708. { "name": "align_corners", "type": "boolean", "default": false },
  709. { "name": "half_pixel_centers", "type": "boolean", "default": false }
  710. ]
  711. },
  712. {
  713. "name": "ReverseSequence",
  714. "attributes": [
  715. { "name": "seq_dim", "type": "int32", "default": 0 },
  716. { "name": "batch_dim", "type": "int32", "default": 0 }
  717. ]
  718. },
  719. {
  720. "name": "RNN",
  721. "category": "Layer",
  722. "attributes": [
  723. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  724. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
  725. ],
  726. "inputs": [
  727. { "name": "X", "type": "T" },
  728. { "name": "W", "type": "T" },
  729. { "name": "R", "type": "T" },
  730. { "name": "b", "type": "T" }
  731. ],
  732. "outputs": [
  733. { "name": "hidden", "type": "T" },
  734. { "name": "output", "type": "T" }
  735. ]
  736. },
  737. {
  738. "name": "Rsqrt",
  739. "inputs": [
  740. { "name": "input", "type": "T" }
  741. ],
  742. "outputs": [
  743. { "name": "output", "type": "T" }
  744. ]
  745. },
  746. {
  747. "name": "Shape",
  748. "attributes": [
  749. { "name": "out_type", "type": "TensorType", "default": "FLOAT32" }
  750. ]
  751. },
  752. {
  753. "name": "Sin",
  754. "inputs": [
  755. { "name": "input" }
  756. ],
  757. "outputs": [
  758. { "name": "output" }
  759. ]
  760. },
  761. {
  762. "name": "SkipGram",
  763. "inputs": [
  764. { "name": "inputs" }
  765. ],
  766. "outputs": [
  767. { "name": "ngrams" }
  768. ],
  769. "attributes": [
  770. { "name": "ngram_size", "type": "int32", "default": 0 },
  771. { "name": "max_skip_size", "type": "int32", "default": 0 },
  772. { "name": "include_all_ngrams", "type": "boolean", "default": false }
  773. ]
  774. },
  775. {
  776. "name": "Slice",
  777. "category": "Tensor",
  778. "inputs": [
  779. { "name": "input" },
  780. { "name": "begin" },
  781. { "name": "size" }
  782. ],
  783. "outputs": [
  784. { "name": "output" }
  785. ]
  786. },
  787. {
  788. "name": "Softmax",
  789. "category": "Activation",
  790. "inputs": [
  791. { "name": "input", "type": "T" }
  792. ],
  793. "outputs": [
  794. { "name": "output", "type": "T" }
  795. ],
  796. "attributes": [
  797. { "name": "beta", "type": "float32", "default": 0 }
  798. ]
  799. },
  800. {
  801. "name": "SpaceToDepth",
  802. "attributes": [
  803. { "name": "block_size", "type": "int32", "default": 0 }
  804. ]
  805. },
  806. {
  807. "name": "SparseToDense",
  808. "attributes": [
  809. { "name": "validate_indices", "type": "boolean", "default": false }
  810. ]
  811. },
  812. {
  813. "name": "Split",
  814. "category": "Tensor",
  815. "inputs": [
  816. { "name": "axis" },
  817. { "name": "input" }
  818. ],
  819. "outputs": [
  820. { "name": "output" }
  821. ],
  822. "attributes": [
  823. { "name": "num_splits", "type": "int32", "default": 0 }
  824. ]
  825. },
  826. {
  827. "name": "SplitV",
  828. "attributes": [
  829. { "name": "num_splits", "type": "int32", "default": 0 }
  830. ]
  831. },
  832. {
  833. "name": "Squeeze",
  834. "category": "Transform",
  835. "inputs": [
  836. { "name": "input" }
  837. ],
  838. "outputs": [
  839. { "name": "output" }
  840. ],
  841. "attributes": [
  842. { "name": "squeeze_dims", "type": "int32[]", "default": 0 }
  843. ]
  844. },
  845. {
  846. "name": "StablehloBroadcastInDim",
  847. "attributes": [
  848. { "name": "broadcast_dimensions", "type": "int64[]", "default": 0 }
  849. ]
  850. },
  851. {
  852. "name": "StablehloCompare",
  853. "attributes": [
  854. { "name": "comparison_direction", "type": "StablehloComparisonDirection", "default": "STABLEHLO_COMPARISON_DIRECTION_EQ" },
  855. { "name": "compare_type", "type": "StablehloComparisonType", "default": "STABLEHLO_COMPARISON_TYPE_NOTYPE" }
  856. ]
  857. },
  858. {
  859. "name": "StablehloConcatenate",
  860. "attributes": [
  861. { "name": "dimension", "type": "int64", "default": 0 }
  862. ]
  863. },
  864. {
  865. "name": "StablehloConvolution",
  866. "attributes": [
  867. { "name": "window_strides", "type": "int64[]", "default": 0 },
  868. { "name": "padding", "type": "int64[]", "default": 0 },
  869. { "name": "lhs_dilation", "type": "int64[]", "default": 0 },
  870. { "name": "rhs_dilation", "type": "int64[]", "default": 0 },
  871. { "name": "window_reversal", "type": "boolean", "default": false },
  872. { "name": "input_batch_dimension", "type": "int64", "default": 0 },
  873. { "name": "input_feature_dimension", "type": "int64", "default": 0 },
  874. { "name": "input_spatial_dimensions", "type": "int64[]", "default": 0 },
  875. { "name": "kernel_input_feature_dimension", "type": "int64", "default": 0 },
  876. { "name": "kernel_output_feature_dimension", "type": "int64", "default": 0 },
  877. { "name": "kernel_spatial_dimensions", "type": "int64[]", "default": 0 },
  878. { "name": "output_batch_dimension", "type": "int64", "default": 0 },
  879. { "name": "output_feature_dimension", "type": "int64", "default": 0 },
  880. { "name": "output_spatial_dimensions", "type": "int64[]", "default": 0 },
  881. { "name": "feature_group_count", "type": "int64", "default": 0 },
  882. { "name": "batch_group_count", "type": "int64", "default": 0 },
  883. { "name": "precision_config", "type": "StablehloPrecisionConfig[]", "default": "DEFAULT" }
  884. ]
  885. },
  886. {
  887. "name": "StablehloCustomCall",
  888. "attributes": [
  889. { "name": "call_target_name", "type": "string", "default": null },
  890. { "name": "has_side_effect", "type": "boolean", "default": false },
  891. { "name": "backend_config", "type": "string", "default": null },
  892. { "name": "api_version", "type": "int32", "default": 0 },
  893. { "name": "called_computations", "type": "int32[]", "default": 0 },
  894. { "name": "custom_attributes", "type": "uint8[]", "default": 0 }
  895. ]
  896. },
  897. {
  898. "name": "StablehloDotGeneral",
  899. "attributes": [
  900. { "name": "lhs_batching_dimensions", "type": "int64[]", "default": 0 },
  901. { "name": "rhs_batching_dimensions", "type": "int64[]", "default": 0 },
  902. { "name": "lhs_contracting_dimensions", "type": "int64[]", "default": 0 },
  903. { "name": "rhs_contracting_dimensions", "type": "int64[]", "default": 0 },
  904. { "name": "precision_config", "type": "StablehloPrecisionConfig[]", "default": "DEFAULT" }
  905. ]
  906. },
  907. {
  908. "name": "StablehloDynamicSlice",
  909. "attributes": [
  910. { "name": "slice_sizes", "type": "int64[]", "default": 0 }
  911. ]
  912. },
  913. {
  914. "name": "StablehloGather",
  915. "attributes": [
  916. { "name": "offset_dims", "type": "int64[]", "default": 0 },
  917. { "name": "collapsed_slice_dims", "type": "int64[]", "default": 0 },
  918. { "name": "start_index_map", "type": "int64[]", "default": 0 },
  919. { "name": "index_vector_dim", "type": "int64", "default": 0 },
  920. { "name": "slice_sizes", "type": "int64[]", "default": 0 },
  921. { "name": "indices_are_sorted", "type": "boolean", "default": false }
  922. ]
  923. },
  924. {
  925. "name": "StablehloIota",
  926. "attributes": [
  927. { "name": "iota_dimension", "type": "int64", "default": 0 }
  928. ]
  929. },
  930. {
  931. "name": "StablehloPad",
  932. "attributes": [
  933. { "name": "edge_padding_low", "type": "int64[]", "default": 0 },
  934. { "name": "edge_padding_high", "type": "int64[]", "default": 0 },
  935. { "name": "interior_padding", "type": "int64[]", "default": 0 }
  936. ]
  937. },
  938. {
  939. "name": "StablehloReduce",
  940. "attributes": [
  941. { "name": "dimensions", "type": "int64[]", "default": 0 },
  942. { "name": "body_subgraph_index", "type": "int32", "default": 0 }
  943. ]
  944. },
  945. {
  946. "name": "StablehloReduceWindow",
  947. "attributes": [
  948. { "name": "window_dimensions", "type": "int64[]", "default": 0 },
  949. { "name": "window_strides", "type": "int64[]", "default": 0 },
  950. { "name": "base_dilations", "type": "int64[]", "default": 0 },
  951. { "name": "window_dilations", "type": "int64[]", "default": 0 },
  952. { "name": "padding", "type": "int64[]", "default": 0 },
  953. { "name": "body_subgraph_index", "type": "int32", "default": 0 }
  954. ]
  955. },
  956. {
  957. "name": "StablehloRngBitGenerator",
  958. "attributes": [
  959. { "name": "algorithm", "type": "RngAlgorithm", "default": "DEFAULT" }
  960. ]
  961. },
  962. {
  963. "name": "StablehloScatter",
  964. "attributes": [
  965. { "name": "indices_are_sorted", "type": "boolean", "default": false },
  966. { "name": "update_window_dims", "type": "int64[]", "default": 0 },
  967. { "name": "inserted_window_dims", "type": "int64[]", "default": 0 },
  968. { "name": "scatter_dims_to_operand_dims", "type": "int64[]", "default": 0 },
  969. { "name": "index_vector_dim", "type": "int64", "default": 0 },
  970. { "name": "unique_indices", "type": "boolean", "default": false },
  971. { "name": "update_computation_subgraph_index", "type": "int32", "default": 0 }
  972. ]
  973. },
  974. {
  975. "name": "StablehloSlice",
  976. "attributes": [
  977. { "name": "start_indices", "type": "int64[]", "default": 0 },
  978. { "name": "limit_indices", "type": "int64[]", "default": 0 },
  979. { "name": "strides", "type": "int64[]", "default": 0 }
  980. ]
  981. },
  982. {
  983. "name": "StablehloSort",
  984. "attributes": [
  985. { "name": "dimension", "type": "int64", "default": 0 },
  986. { "name": "is_stable", "type": "boolean", "default": false },
  987. { "name": "comparator_subgraph_index", "type": "int32", "default": 0 }
  988. ]
  989. },
  990. {
  991. "name": "StablehloTranspose",
  992. "attributes": [
  993. { "name": "permutation", "type": "int64[]", "default": 0 }
  994. ]
  995. },
  996. {
  997. "name": "StablehloWhile",
  998. "attributes": [
  999. { "name": "cond_subgraph_index", "type": "int32", "default": 0 },
  1000. { "name": "body_subgraph_index", "type": "int32", "default": 0 }
  1001. ]
  1002. },
  1003. {
  1004. "name": "StridedSlice",
  1005. "category": "Tensor",
  1006. "inputs": [
  1007. { "name": "input" },
  1008. { "name": "begin" },
  1009. { "name": "end" },
  1010. { "name": "strides" }
  1011. ],
  1012. "outputs": [
  1013. { "name": "output" }
  1014. ],
  1015. "attributes": [
  1016. { "name": "begin_mask", "type": "int32", "default": 0 },
  1017. { "name": "end_mask", "type": "int32", "default": 0 },
  1018. { "name": "ellipsis_mask", "type": "int32", "default": 0 },
  1019. { "name": "new_axis_mask", "type": "int32", "default": 0 },
  1020. { "name": "shrink_axis_mask", "type": "int32", "default": 0 },
  1021. { "name": "offset", "type": "boolean", "default": false }
  1022. ]
  1023. },
  1024. {
  1025. "name": "Sub",
  1026. "attributes": [
  1027. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  1028. { "name": "pot_scale_int16", "type": "boolean", "default": true }
  1029. ],
  1030. "inputs": [
  1031. { "name": "A", "type": "T" },
  1032. { "name": "B", "type": "T" }
  1033. ],
  1034. "outputs": [
  1035. { "name": "C", "type": "T" }
  1036. ]
  1037. },
  1038. {
  1039. "name": "Sum",
  1040. "attributes": [
  1041. { "name": "keep_dims", "type": "boolean" }
  1042. ],
  1043. "inputs": [
  1044. { "name": "input", "type": "T" },
  1045. { "name": "axis", "type": "T" }
  1046. ],
  1047. "outputs": [
  1048. { "name": "output", "type": "T" }
  1049. ]
  1050. },
  1051. {
  1052. "name": "SVDF",
  1053. "category": "Layer",
  1054. "attributes": [
  1055. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  1056. { "name": "rank", "type": "int32", "default": 0 },
  1057. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
  1058. ],
  1059. "inputs": [
  1060. { "name": "input", "type": "T" },
  1061. { "name": "feature", "type": "T" },
  1062. { "name": "time", "type": "T" },
  1063. { "name": "bias", "type": "T" },
  1064. { "name": "state", "type": "T" }
  1065. ],
  1066. "outputs": [
  1067. { "name": "state_out", "type": "T" },
  1068. { "name": "output", "type": "T" }
  1069. ]
  1070. },
  1071. {
  1072. "name": "Tanh",
  1073. "category": "Activation",
  1074. "inputs": [
  1075. { "name": "input", "type": "T" }
  1076. ],
  1077. "outputs": [
  1078. { "name": "output", "type": "T" }
  1079. ]
  1080. },
  1081. {
  1082. "name": "Tile",
  1083. "inputs": [
  1084. { "name": "input" },
  1085. { "name": "multipliers" }
  1086. ],
  1087. "outputs": [
  1088. { "name": "output" }
  1089. ]
  1090. },
  1091. {
  1092. "name": "Transpose",
  1093. "category": "Transform",
  1094. "inputs": [
  1095. { "name": "input" },
  1096. { "name": "perm" }
  1097. ],
  1098. "outputs": [
  1099. { "name": "output" }
  1100. ]
  1101. },
  1102. {
  1103. "name": "TransposeConv",
  1104. "category": "Layer",
  1105. "attributes": [
  1106. { "name": "padding", "type": "Padding", "default": "SAME" },
  1107. { "name": "stride_w", "type": "int32", "default": 0 },
  1108. { "name": "stride_h", "type": "int32", "default": 0 },
  1109. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  1110. { "name": "quantized_bias_type", "type": "TensorType", "default": "FLOAT32" }
  1111. ],
  1112. "inputs": [
  1113. { "name": "output_shape" },
  1114. { "name": "weights" },
  1115. { "name": "input" },
  1116. { "name": "bias", "optional": true }
  1117. ],
  1118. "outputs": [
  1119. { "name": "output" }
  1120. ]
  1121. },
  1122. {
  1123. "name": "UnidirectionalSequenceLSTM",
  1124. "category": "Layer",
  1125. "attributes": [
  1126. { "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
  1127. { "name": "cell_clip", "type": "float32", "default": 0 },
  1128. { "name": "proj_clip", "type": "float32", "default": 0 },
  1129. { "name": "time_major", "type": "boolean", "default": false },
  1130. { "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false },
  1131. { "name": "diagonal_recurrent_tensors", "type": "boolean", "default": false }
  1132. ],
  1133. "inputs": [
  1134. { "name": "input", "type": "T", "description": "Input tensor." },
  1135. { "name": "input_input_weights", "type": "T", "option": "optional", "description": "Input to input weights tensor.", "visible": false },
  1136. { "name": "input_forget_weights", "type": "T", "description": "Input to forget weights tensor.", "visible": false },
  1137. { "name": "input_cell_weights", "type": "T", "description": "Input to cell weights tensor.", "visible": false },
  1138. { "name": "input_output_weights", "type": "T", "description": "Input to output weights tensor.", "visible": false },
  1139. { "name": "recurrent_input_weights", "type": "T", "option": "optional", "description": "Recurrent to input weights tensor.", "visible": false },
  1140. { "name": "recurrent_forget_weights", "type": "T", "description": "Recurrent to forget weights tensor.", "visible": false },
  1141. { "name": "recurrent_cell_weights", "type": "T", "description": "Recurrent to cell weights tensor.", "visible": false },
  1142. { "name": "recurrent_output_weights", "type": "T", "description": "Recurrent to output weights tensor.", "visible": false },
  1143. { "name": "cell_input_weights", "type": "T", "option": "optional", "description": "Cell to input weights tensor.", "visible": false },
  1144. { "name": "cell_forget_weights", "type": "T", "option": "optional", "description": "Cell to forget weights tensor.", "visible": false },
  1145. { "name": "cell_output_weights", "type": "T", "option": "optional", "description": "Cell to output weights tensor.", "visible": false },
  1146. { "name": "input_gate_bias", "type": "T", "option": "optional", "description": "Input gate bias tensor.", "visible": false },
  1147. { "name": "forget_gate_bias", "type": "T", "description": "Forget gate bias tensor.", "visible": false },
  1148. { "name": "cell_gate_bias", "type": "T", "description": "Cell gate bias tensor.", "visible": false },
  1149. { "name": "output_gate_bias", "type": "T", "description": "Output gate bias tensor.", "visible": false },
  1150. { "name": "projection_weights", "type": "T", "option": "optional", "description": "Projection weights tensor.", "visible": false },
  1151. { "name": "projection_bias", "type": "T", "option": "optional", "description": "Projection bias tensor.", "visible": false },
  1152. { "name": "output_state_in", "type": "T", "option": "optional", "visible": false },
  1153. { "name": "cell_state_in", "type": "T", "option": "optional", "visible": false },
  1154. { "name": "activation", "type": "T", "option": "optional", "visible": false },
  1155. { "name": "cell_clip", "type": "T", "option": "optional", "visible": false },
  1156. { "name": "proj_clip", "type": "T", "option": "optional", "visible": false },
  1157. { "name": "time_major", "type": "T", "option": "optional", "visible": false },
  1158. { "name": "input_layer_norm_weights", "type": "T", "option": "optional", "visible": false },
  1159. { "name": "forget_layer_norm_weights", "type": "T", "option": "optional", "visible": false },
  1160. { "name": "cell_layer_norm_weights", "type": "T", "option": "optional", "visible": false },
  1161. { "name": "output_layer_norm_weights", "type": "T", "option": "optional", "visible": false }
  1162. ]
  1163. },
  1164. {
  1165. "name": "Unique",
  1166. "attributes": [
  1167. { "name": "idx_out_type", "type": "TensorType", "default": "INT32" }
  1168. ]
  1169. },
  1170. {
  1171. "name": "Unpack",
  1172. "attributes": [
  1173. { "name": "num", "type": "int32", "default": 0 },
  1174. { "name": "axis", "type": "int32", "default": 0 }
  1175. ]
  1176. },
  1177. {
  1178. "name": "UnsortedSegmentProd",
  1179. "attributes": [
  1180. { "name": "num_segments", "type": "int32", "default": 0 }
  1181. ]
  1182. },
  1183. {
  1184. "name": "VarHandle",
  1185. "attributes": [
  1186. { "name": "container", "type": "string", "default": null },
  1187. { "name": "shared_name", "type": "string", "default": null }
  1188. ]
  1189. },
  1190. {
  1191. "name": "While",
  1192. "attributes": [
  1193. { "name": "cond_subgraph_index", "type": "int32", "default": 0 },
  1194. { "name": "body_subgraph_index", "type": "int32", "default": 0 }
  1195. ]
  1196. }
  1197. ]