circle-metadata.json 36 KB

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