tflite-metadata.json 36 KB

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