tengine-metadata.json 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933
  1. [
  2. {
  3. "name": "Absval",
  4. "category": "Data"
  5. },
  6. {
  7. "name": "Accuracy"
  8. },
  9. {
  10. "name": "Addn",
  11. "attributes": [
  12. { "name": "axis", "type": "int32", "default": 0 }
  13. ]
  14. },
  15. {
  16. "name": "ArgMax",
  17. "category": "Layer",
  18. "attributes": [
  19. { "name": "axis", "type": "int32", "default": 0 }
  20. ]
  21. },
  22. {
  23. "name": "ArgMin",
  24. "category": "Layer",
  25. "attributes": [
  26. { "name": "axis", "type": "int32", "default": 0 }
  27. ]
  28. },
  29. {
  30. "name": "BatchNormalization",
  31. "category": "Normalization",
  32. "attributes": [
  33. { "name": "rescale_factor", "type": "float32", "default": 1 },
  34. { "name": "eps", "type": "float32", "default": 0.00001 },
  35. { "name": "caffe_flavor", "type": "int32", "default": 0 }
  36. ],
  37. "inputs": [
  38. { "name": "input" },
  39. { "name": "gamma" },
  40. { "name": "beta" },
  41. { "name": "mean" },
  42. { "name": "var" }
  43. ]
  44. },
  45. {
  46. "name": "BatchToSpaceND",
  47. "category": "Shape",
  48. "attributes": [
  49. { "name": "dilation_x", "type": "int32", "default": 0 },
  50. { "name": "dilation_y", "type": "int32", "default": 0 },
  51. { "name": "crop_top", "type": "int32", "default": 0 },
  52. { "name": "crop_bottom", "type": "int32", "default": 0 },
  53. { "name": "crop_left", "type": "int32", "default": 0 },
  54. { "name": "crop_right", "type": "int32", "default": 0 }
  55. ]
  56. },
  57. {
  58. "name": "Bias",
  59. "category": "Layer",
  60. "attributes": [
  61. { "name": "bias_size", "type": "int32", "default": 0 }
  62. ]
  63. },
  64. {
  65. "name": "BilinearResize",
  66. "category": "Shape",
  67. "attributes": [
  68. { "name": "scale_x", "type": "float32", "default": 0 },
  69. { "name": "scale_y", "type": "float32", "default": 0 },
  70. { "name": "type", "type": "int32", "default": 0 }
  71. ]
  72. },
  73. {
  74. "name": "BroadMul",
  75. "category": "Layer"
  76. },
  77. {
  78. "name": "Cast",
  79. "attributes": [
  80. { "name": "type_from", "type": "int32", "default": 0 },
  81. { "name": "type_to", "type": "int32", "default": 0 }
  82. ]
  83. },
  84. {
  85. "name": "Ceil",
  86. "category": "Layer"
  87. },
  88. {
  89. "name": "Clip",
  90. "category": "Layer",
  91. "attributes": [
  92. { "name": "max", "type": "float32", "default": 0 },
  93. { "name": "min", "type": "float32", "default": 0 }
  94. ]
  95. },
  96. {
  97. "name": "Comparison",
  98. "category": "Layer",
  99. "attributes": [
  100. { "name": "type", "type": "int32", "default": 0 }
  101. ]
  102. },
  103. {
  104. "name": "Concat",
  105. "category": "Shape",
  106. "attributes": [
  107. { "name": "axis", "type": "int32", "default": 1 }
  108. ],
  109. "inputs": [
  110. { "name": "inputs", "option": "variadic" }
  111. ]
  112. },
  113. {
  114. "name": "Const"
  115. },
  116. {
  117. "name": "Convolution",
  118. "category": "Layer",
  119. "attributes": [
  120. { "name": "kernel_h", "type": "int32", "default": 1 },
  121. { "name": "kernel_w", "type": "int32", "default": 1 },
  122. { "name": "stride_h", "type": "int32", "default": 1 },
  123. { "name": "stride_w", "type": "int32", "default": 1 },
  124. { "name": "dilation_h", "type": "int32", "default": 1, "visible": false },
  125. { "name": "dilation_w", "type": "int32", "default": 1, "visible": false },
  126. { "name": "input_channel", "type": "int32", "default": 1 },
  127. { "name": "output_channel", "type": "int32", "default": 1 },
  128. { "name": "group", "type": "int32", "default": 1, "visible": false },
  129. { "name": "activation", "type": "int32", "default": -1 },
  130. { "name": "pad_h0", "type": "int32", "default": 0, "visible": false },
  131. { "name": "pad_w0", "type": "int32", "default": 0, "visible": false },
  132. { "name": "pad_h1", "type": "int32", "default": 0, "visible": false },
  133. { "name": "pad_w1", "type": "int32", "default": 0, "visible": false }
  134. ],
  135. "inputs": [
  136. { "name": "input" },
  137. { "name": "weight" },
  138. { "name": "bias" }
  139. ]
  140. },
  141. {
  142. "name": "Crop",
  143. "category": "Shape",
  144. "attributes": [
  145. { "name": "num_args", "type": "int32", "default": 0 },
  146. { "name": "offset_c", "type": "int32", "default": 0 },
  147. { "name": "offset_h", "type": "int32", "default": 0 },
  148. { "name": "offset_w", "type": "int32", "default": 0 },
  149. { "name": "crop_h", "type": "int32", "default": 0 },
  150. { "name": "crop_w", "type": "int32", "default": 0 },
  151. { "name": "center_crop", "type": "bool", "default": 0 },
  152. { "name": "axis", "type": "int32", "default": 0 },
  153. { "name": "flag", "type": "int32", "default": 0 }
  154. ]
  155. },
  156. {
  157. "name": "Deconvolution",
  158. "category": "Layer",
  159. "attributes": [
  160. { "name": "num_output", "type": "int32", "default": 1 },
  161. { "name": "kernel_h", "type": "int32", "default": 1 },
  162. { "name": "kernel_w", "type": "int32", "default": 1 },
  163. { "name": "stride_h", "type": "int32", "default": 1 },
  164. { "name": "stride_w", "type": "int32", "default": 1 },
  165. { "name": "pad_w0", "type": "int32", "default": 0 },
  166. { "name": "pad_h0", "type": "int32", "default": 0 },
  167. { "name": "pad_w1", "type": "int32", "default": 0 },
  168. { "name": "pad_h1", "type": "int32", "default": 0 },
  169. { "name": "dilation_h", "type": "int32", "default": 1 },
  170. { "name": "dilation_w", "type": "int32", "default": 1 },
  171. { "name": "group", "type": "int32", "default": 1 },
  172. { "name": "activation", "type": "int32", "default": -1 }
  173. ],
  174. "inputs": [
  175. { "name": "input" },
  176. { "name": "weight" },
  177. { "name": "bias" }
  178. ]
  179. },
  180. {
  181. "name": "DepthToSpace",
  182. "category": "Shape",
  183. "attributes": [
  184. { "name": "block_size", "type": "int32", "default": 0 }
  185. ]
  186. },
  187. {
  188. "name": "DetectionOutput",
  189. "category": "Layer",
  190. "attributes": [
  191. { "name": "num_classes", "type": "int32", "default": 0 },
  192. { "name": "keep_top_k", "type": "int32", "default": 0 },
  193. { "name": "nms_top_k", "type": "int32", "default": 0 },
  194. { "name": "confidence_threshold", "type": "float32", "default": 0 },
  195. { "name": "nms_threshold", "type": "float32", "default": 0 }
  196. ]
  197. },
  198. {
  199. "name": "DetectionPostProcess",
  200. "category": "Layer",
  201. "attributes": [
  202. { "name": "max_detections", "type": "int32", "default": 0 },
  203. { "name": "max_classes_per_detection", "type": "int32", "default": 0 },
  204. { "name": "nms_score_threshold", "type": "float32", "default": 0 },
  205. { "name": "nms_iou_threshold", "type": "float32", "default": 0 },
  206. { "name": "num_classes", "type": "int32", "default": 0 },
  207. { "name": "scales", "type": "float32[]", "default": [] }
  208. ],
  209. "inputs": [
  210. { "name": "input" },
  211. { "name": "score" },
  212. { "name": "anchor" }
  213. ],
  214. "outputs": [
  215. { "name": "detect_boxes" },
  216. { "name": "detect_classes" },
  217. { "name": "detect_scores" },
  218. { "name": "detect_num" }
  219. ]
  220. },
  221. {
  222. "name": "DropOut",
  223. "category": "Dropout"
  224. },
  225. {
  226. "name": "Eltwise",
  227. "attributes": [
  228. { "name": "type", "type": "uint32", "default": 0 },
  229. { "name": "caffe_flavor", "type": "int32", "default": 0 }
  230. ],
  231. "inputs": [
  232. { "name": "inputs", "option": "variadic" }
  233. ]
  234. },
  235. {
  236. "name": "ELU",
  237. "category": "Activation",
  238. "attributes": [
  239. { "name": "alpha", "type": "float32", "default": 0 }
  240. ]
  241. },
  242. {
  243. "name": "Embed",
  244. "category": "Transform",
  245. "attributes": [
  246. { "name": "num_output", "type": "int32", "default": 0 },
  247. { "name": "input_dim", "type": "int32", "default": 0 },
  248. { "name": "bias_term", "type": "int32", "default": 0 },
  249. { "name": "weight_data_size", "type": "int32", "default": 0 }
  250. ]
  251. },
  252. {
  253. "name": "Expand",
  254. "category": "Layer",
  255. "attributes": [
  256. { "name": "v_shape[]", "type": "int32[]", "default": [] }
  257. ]
  258. },
  259. {
  260. "name": "Expanddims",
  261. "category": "Shape",
  262. "attributes": [
  263. { "name": "axis", "type": "int32", "default": 0 }
  264. ]
  265. },
  266. {
  267. "name": "Flatten",
  268. "category": "Shape",
  269. "attributes": [
  270. { "name": "axis", "type": "int32", "default": 0 },
  271. { "name": "end_axis", "type": "int32", "default": 0 }
  272. ]
  273. },
  274. {
  275. "name": "FullyConnected",
  276. "category": "Layer",
  277. "attributes": [
  278. { "name": "num_output", "type": "int32", "default": 10 }
  279. ],
  280. "inputs": [
  281. { "name": "input" },
  282. { "name": "weight" },
  283. { "name": "bias" }
  284. ]
  285. },
  286. {
  287. "name": "FusedbnScaleRelu",
  288. "category": "Activation"
  289. },
  290. {
  291. "name": "Gather",
  292. "category": "Transform",
  293. "attributes": [
  294. { "name": "axis", "type": "int32", "default": 0 },
  295. { "name": "indices_num", "type": "int32", "default": 0 }
  296. ]
  297. },
  298. {
  299. "name": "Gemm",
  300. "attributes": [
  301. { "name": "alpha", "type": "float32", "default": 0 },
  302. { "name": "beta", "type": "float32", "default": 0 },
  303. { "name": "transA", "type": "int32", "default": 0 },
  304. { "name": "transB", "type": "int32", "default": 0 }
  305. ]
  306. },
  307. {
  308. "name": "Generic",
  309. "attributes": [
  310. { "name": "max_input_num", "type": "int32", "default": 0 },
  311. { "name": "max_output_num", "type": "int32", "default": 0 },
  312. { "name": "opname", "type": "string", "default": "" }
  313. ]
  314. },
  315. {
  316. "name": "GRU",
  317. "category": "Layer",
  318. "attributes": [
  319. { "name": "clip", "type": "float32", "default": 0 },
  320. { "name": "output_len", "type": "int32", "default": 0 },
  321. { "name": "sequence_len", "type": "int32", "default": 0 },
  322. { "name": "input_size", "type": "int32", "default": 0 },
  323. { "name": "hidden_size", "type": "int32", "default": 0 },
  324. { "name": "has_clip", "type": "int32", "default": 0 },
  325. { "name": "has_gate_bias", "type": "int32", "default": 0 },
  326. { "name": "has_candidate_bias", "type": "int32", "default": 0 },
  327. { "name": "has_init_state", "type": "int32", "default": 0 },
  328. { "name": "mxnet_flag", "type": "int32", "default": 0 }
  329. ]
  330. },
  331. {
  332. "name": "Hardsigmoid",
  333. "category": "Activation",
  334. "attributes": [
  335. { "name": "alpha", "type": "float32", "default": 0 },
  336. { "name": "beta", "type": "float32", "default": 0 }
  337. ]
  338. },
  339. {
  340. "name": "HardSwish",
  341. "category": "Activation",
  342. "attributes": [
  343. { "name": "alpha", "type": "float32", "default": 0 },
  344. { "name": "beta", "type": "float32", "default": 0 }
  345. ]
  346. },
  347. {
  348. "name": "InstanceNorm",
  349. "category": "Normalization",
  350. "attributes": [
  351. { "name": "eps", "type": "float32", "default": 0 }
  352. ]
  353. },
  354. {
  355. "name": "Interp",
  356. "category": "Layer",
  357. "attributes": [
  358. { "name": "resize_type", "type": "int32", "default": 0 },
  359. { "name": "width_scale", "type": "float32", "default": 0 },
  360. { "name": "height_scale", "type": "float32", "default": 0 },
  361. { "name": "output_width", "type": "int32", "default": 0 },
  362. { "name": "output_height", "type": "int32", "default": 0 }
  363. ]
  364. },
  365. {
  366. "name": "L2Normalization",
  367. "category": "Layer"
  368. },
  369. {
  370. "name": "L2Pool",
  371. "category": "Layer"
  372. },
  373. {
  374. "name": "Logical",
  375. "category": "Layer",
  376. "attributes": [
  377. { "name": "type", "type": "int32", "default": 0 }
  378. ]
  379. },
  380. {
  381. "name": "Logistic",
  382. "category": "Activation"
  383. },
  384. {
  385. "name": "LogSoftmax",
  386. "category": "Layer"
  387. },
  388. {
  389. "name": "LRN",
  390. "category": "Normalization",
  391. "attributes": [
  392. { "name": "local_size", "type": "int32", "default": 0 },
  393. { "name": "alpha", "type": "float32", "default": 0 },
  394. { "name": "beta", "type": "float32", "default": 0 },
  395. { "name": "norm_region", "type": "int32", "default": 0 },
  396. { "name": "k", "type": "float32", "default": 0 }
  397. ]
  398. },
  399. {
  400. "name": "LSTM",
  401. "category": "Layer",
  402. "attributes": [
  403. { "name": "forget_bias", "type": "float32", "default": 0 },
  404. { "name": "clip", "type": "float32", "default": 0 },
  405. { "name": "output_len", "type": "int32", "default": 0 },
  406. { "name": "sequence_len", "type": "int32", "default": 0 },
  407. { "name": "input_size", "type": "int32", "default": 0 },
  408. { "name": "hidden_size", "type": "int32", "default": 0 },
  409. { "name": "cell_size", "type": "int32", "default": 0 },
  410. { "name": "has_peephole", "type": "int32", "default": 0 },
  411. { "name": "has_projection", "type": "int32", "default": 0 },
  412. { "name": "has_clip", "type": "int32", "default": 0 },
  413. { "name": "has_bias", "type": "int32", "default": 0 },
  414. { "name": "has_init_state", "type": "int32", "default": 0 },
  415. { "name": "forget_act", "type": "int32", "default": 0 },
  416. { "name": "input_act", "type": "int32", "default": 0 },
  417. { "name": "output_act", "type": "int32", "default": 0 },
  418. { "name": "cellin_act", "type": "int32", "default": 0 },
  419. { "name": "cellout_act", "type": "int32", "default": 0 },
  420. { "name": "mxnet_flag", "type": "int32", "default": 0 }
  421. ]
  422. },
  423. {
  424. "name": "MatMul",
  425. "category": "Layer"
  426. },
  427. {
  428. "name": "Max",
  429. "category": "Layer"
  430. },
  431. {
  432. "name": "Mean",
  433. "category": "Layer"
  434. },
  435. {
  436. "name": "Min",
  437. "category": "Layer"
  438. },
  439. {
  440. "name": "Mish",
  441. "category": "Activation"
  442. },
  443. {
  444. "name": "MVN",
  445. "attributes": [
  446. { "name": "across_channels", "type": "int32", "default": 0 },
  447. { "name": "normalize_variance", "type": "int32", "default": 0 },
  448. { "name": "eps", "type": "float32", "default": 0 }
  449. ]
  450. },
  451. {
  452. "name": "Noop",
  453. "category": "Layer"
  454. },
  455. {
  456. "name": "Normalize",
  457. "category": "Normalization",
  458. "attributes": [
  459. { "name": "across_spatial", "type": "int32", "default": 0 },
  460. { "name": "channel_shared", "type": "int32", "default": 0 }
  461. ]
  462. },
  463. {
  464. "name": "Num",
  465. "category": "Layer"
  466. },
  467. {
  468. "name": "PackModel",
  469. "category": "Layer",
  470. "attributes": [
  471. { "name": "buffer size", "type": "int32", "default": 1 },
  472. { "name": "model version", "type": "int32", "default": 1 }
  473. ],
  474. "inputs": [
  475. { "name": "input" },
  476. { "name": "Source Model" }
  477. ]
  478. },
  479. {
  480. "name": "Pad",
  481. "category": "Tensor",
  482. "attributes": [
  483. { "name": "pad_n_0", "type": "int32", "default": -1 },
  484. { "name": "pad_n_1", "type": "int32", "default": -1 },
  485. { "name": "pad_c_0", "type": "int32", "default": -1 },
  486. { "name": "pad_c_1", "type": "int32", "default": -1 },
  487. { "name": "pad_h_0", "type": "int32", "default": -1 },
  488. { "name": "pad_h_1", "type": "int32", "default": -1 },
  489. { "name": "pad_w_0", "type": "int32", "default": -1 },
  490. { "name": "pad_w_1", "type": "int32", "default": -1 },
  491. { "name": "mode", "type": "int32", "default": 0 },
  492. { "name": "value", "type": "float32", "default": 0 }
  493. ],
  494. "inputs": [
  495. { "name": "input" },
  496. { "name": "paddings" }
  497. ]
  498. },
  499. {
  500. "name": "Permute",
  501. "category": "Shape",
  502. "attributes": [
  503. { "name": "flag", "type": "int32", "default": 0 },
  504. { "name": "order0", "type": "int32", "default": 0 },
  505. { "name": "order1", "type": "int32", "default": 0 },
  506. { "name": "order2", "type": "int32", "default": 0 },
  507. { "name": "order3", "type": "int32", "default": 0 }
  508. ]
  509. },
  510. {
  511. "name": "Pooling",
  512. "category": "Pool",
  513. "attributes": [
  514. { "name": "alg", "type": "int32", "default": 0 },
  515. { "name": "kernel_h", "type": "int32", "default": 0 },
  516. { "name": "kernel_w", "type": "int32", "default": 0 },
  517. { "name": "stride_h", "type": "int32", "default": 0 },
  518. { "name": "stride_w", "type": "int32", "default": 0 },
  519. { "name": "global", "type": "int32", "default": 0 },
  520. { "name": "caffe_flavor", "type": "int32", "default": 0 },
  521. { "name": "pad_h0", "type": "int32", "default": 0 },
  522. { "name": "pad_w0", "type": "int32", "default": 0 },
  523. { "name": "pad_h1", "type": "int32", "default": 0 },
  524. { "name": "pad_w1", "type": "int32", "default": 0 }
  525. ]
  526. },
  527. {
  528. "name": "Prelu",
  529. "category": "Activation",
  530. "inputs": [
  531. { "name": "input" },
  532. { "name": "slope" }
  533. ]
  534. },
  535. {
  536. "name": "PriorBox",
  537. "attributes": [
  538. { "name": "min_size", "type": "float32[]", "default": [] },
  539. { "name": "max_size", "type": "float32[]", "default": [] },
  540. { "name": "variance", "type": "float32[]", "default": [] },
  541. { "name": "aspect_ratio", "type": "float32[]", "default": [] },
  542. { "name": "flip", "type": "int32", "default": 0 },
  543. { "name": "clip", "type": "int32", "default": 0 },
  544. { "name": "img_size", "type": "int32", "default": 0 },
  545. { "name": "img_h", "type": "int32", "default": 0 },
  546. { "name": "img_w", "type": "int32", "default": 0 },
  547. { "name": "step_w", "type": "float32", "default": 0 },
  548. { "name": "step_h", "type": "float32", "default": 0 },
  549. { "name": "offset", "type": "float32", "default": 0 },
  550. { "name": "num_priors", "type": "int32", "default": 0 },
  551. { "name": "out_dim", "type": "int32", "default": 0 }
  552. ]
  553. },
  554. {
  555. "name": "Psroipooling",
  556. "category": "Pool",
  557. "attributes": [
  558. { "name": "pooled_w", "type": "int32", "default": 0 },
  559. { "name": "pooled_h", "type": "int32", "default": 0 },
  560. { "name": "spatial_scale", "type": "float32", "default": 0 },
  561. { "name": "output_dim", "type": "int32", "default": 0 }
  562. ]
  563. },
  564. {
  565. "name": "ReduceL2",
  566. "category": "Layer",
  567. "attributes": [
  568. { "name": "axis", "type": "int32", "default": 0 },
  569. { "name": "keepdim", "type": "int32", "default": 0 }
  570. ]
  571. },
  572. {
  573. "name": "Reduction",
  574. "attributes": [
  575. { "name": "dim_0", "type": "int32", "default": -2 },
  576. { "name": "dim_1", "type": "int32", "default": -2 },
  577. { "name": "dim_2", "type": "int32", "default": -2 },
  578. { "name": "dim_3", "type": "int32", "default": -2 },
  579. { "name": "type", "type": "int32", "default": 0 },
  580. { "name": "keepdim", "type": "int32", "default": 0 }
  581. ]
  582. },
  583. {
  584. "name": "Region",
  585. "attributes": [
  586. { "name": "num_classes", "type": "int32", "default": 0 },
  587. { "name": "side", "type": "int32", "default": 0 },
  588. { "name": "num_box", "type": "int32", "default": 0 },
  589. { "name": "coords", "type": "int32", "default": 0 },
  590. { "name": "confidence_threshold", "type": "float32", "default": 0 },
  591. { "name": "nms_threshold", "type": "float32", "default": 0 },
  592. { "name": "biases", "type": "float32[]", "default": [] }
  593. ]
  594. },
  595. {
  596. "name": "ReLU",
  597. "category": "Activation",
  598. "attributes": [
  599. { "name": "negative_slope", "type": "float32", "default": 0 }
  600. ]
  601. },
  602. {
  603. "name": "Relu1",
  604. "category": "Layer"
  605. },
  606. {
  607. "name": "ReLU6",
  608. "category": "Activation"
  609. },
  610. {
  611. "name": "Reorg",
  612. "category": "Shape",
  613. "attributes": [
  614. { "name": "stride", "type": "int32", "default": 0 }
  615. ]
  616. },
  617. {
  618. "name": "Reshape",
  619. "category": "Shape",
  620. "attributes": [
  621. { "name": "is_mxnet", "type": "int32", "default": 0 },
  622. { "name": "reverse", "type": "int32", "default": 0 },
  623. { "name": "shape", "type": "int32[]", "default": [] }
  624. ],
  625. "inputs": [
  626. { "name": "input" },
  627. { "name": "shape" }
  628. ]
  629. },
  630. {
  631. "name": "Reshape",
  632. "category": "Shape",
  633. "attributes": [
  634. { "name": "dim_0", "type": "int32", "default": 0 },
  635. { "name": "dim_1", "type": "int32", "default": 0 },
  636. { "name": "dim_2", "type": "int32", "default": 0 },
  637. { "name": "dim_3", "type": "int32", "default": 0 },
  638. { "name": "dim_size", "type": "int32", "default": 0 },
  639. { "name": "axis", "type": "int32", "default": 0 }
  640. ],
  641. "inputs": [
  642. { "name": "input" },
  643. { "name": "shape" }
  644. ]
  645. },
  646. {
  647. "name": "Reshape",
  648. "version": 2,
  649. "category": "Shape",
  650. "attributes": [
  651. { "name": "is_mxnet", "type": "int32", "default": 0 },
  652. { "name": "reverse", "type": "int32", "default": 0 },
  653. { "name": "shape", "type": "int32[]", "default": [] }
  654. ],
  655. "inputs": [
  656. { "name": "input" },
  657. { "name": "shape" }
  658. ]
  659. },
  660. {
  661. "name": "Resize",
  662. "category": "Data",
  663. "attributes": [
  664. { "name": "scale_x", "type": "float32", "default": 0 },
  665. { "name": "scale_y", "type": "float32", "default": 0 },
  666. { "name": "type", "type": "int32", "default": 0 }
  667. ]
  668. },
  669. {
  670. "name": "Reverse",
  671. "category": "Shape"
  672. },
  673. {
  674. "name": "RNN",
  675. "category": "Layer",
  676. "attributes": [
  677. { "name": "clip", "type": "float32", "default": 0 },
  678. { "name": "output_len", "type": "int32", "default": 0 },
  679. { "name": "sequence_len", "type": "int32", "default": 0 },
  680. { "name": "input_size", "type": "int32", "default": 0 },
  681. { "name": "hidden_size", "type": "int32", "default": 0 },
  682. { "name": "has_clip", "type": "int32", "default": 0 },
  683. { "name": "has_bias", "type": "int32", "default": 0 },
  684. { "name": "has_init_state", "type": "int32", "default": 0 },
  685. { "name": "activation", "type": "int32", "default": 0 }
  686. ]
  687. },
  688. {
  689. "name": "ROIAlign",
  690. "attributes": [
  691. { "name": "pooled_width", "type": "int32", "default": 0 },
  692. { "name": "pooled_height", "type": "int32", "default": 0 },
  693. { "name": "spatial_scale", "type": "float32", "default": 0 }
  694. ]
  695. },
  696. {
  697. "name": "RoiPooling",
  698. "category": "Pool",
  699. "attributes": [
  700. { "name": "pooled_h", "type": "int32", "default": 0 },
  701. { "name": "pooled_w", "type": "int32", "default": 0 },
  702. { "name": "spatial_scale", "type": "float32", "default": 0 }
  703. ]
  704. },
  705. {
  706. "name": "Round",
  707. "category": "Layer"
  708. },
  709. {
  710. "name": "RPN",
  711. "attributes": [
  712. { "name": "ratios", "type": "float32[]", "default": [] },
  713. { "name": "anchor_scales", "type": "float32[]", "default": [] },
  714. { "name": "feat_stride", "type": "int32", "default": 0 },
  715. { "name": "basesize", "type": "int32", "default": 0 },
  716. { "name": "min_size", "type": "int32", "default": 0 },
  717. { "name": "per_nms_topn", "type": "int32", "default": 0 },
  718. { "name": "post_nms_topn", "type": "int32", "default": 0 },
  719. { "name": "nms_thresh", "type": "float32", "default": 0 },
  720. { "name": "anchors", "default": 0 }
  721. ]
  722. },
  723. {
  724. "name": "Scale",
  725. "category": "Layer",
  726. "attributes": [
  727. { "name": "axis", "type": "int32", "default": 0 },
  728. { "name": "num_axes", "type": "int32", "default": 0 },
  729. { "name": "bias_term", "type": "int32", "default": 0 }
  730. ],
  731. "inputs": [
  732. { "name": "input" },
  733. { "name": "gamma" },
  734. { "name": "beta" }
  735. ]
  736. },
  737. {
  738. "name": "Scatter",
  739. "category": "Layer",
  740. "attributes": [
  741. { "name": "axis", "type": "int32", "default": 0 },
  742. { "name": "is_onnx", "type": "boolean", "default": false }
  743. ]
  744. },
  745. {
  746. "name": "SELU",
  747. "category": "Activation",
  748. "attributes": [
  749. { "name": "alpha", "type": "float32", "default": 0 },
  750. { "name": "lambda", "type": "float32", "default": 0 }
  751. ]
  752. },
  753. {
  754. "name": "Shape",
  755. "category": "Shape"
  756. },
  757. {
  758. "name": "ShuffleChannel",
  759. "category": "shape",
  760. "attributes": [
  761. { "name": "group", "type": "int32", "default": 0 }
  762. ]
  763. },
  764. {
  765. "name": "Sigmoid",
  766. "category": "Activation"
  767. },
  768. {
  769. "name": "Slice",
  770. "category": "Shape",
  771. "attributes": [
  772. { "name": "axis", "type": "int32", "default": 0 },
  773. { "name": "slice_points", "type": "int32[]", "default": [] },
  774. { "name": "begins", "type": "int32[]", "default": [] },
  775. { "name": "sizes", "type": "int32[]", "default": [] },
  776. { "name": "iscaffe", "type": "int32", "default": 0 },
  777. { "name": "ismxnet", "type": "int32", "default": 0 },
  778. { "name": "isonnx", "type": "int32", "default": 0 },
  779. { "name": "begin", "type": "int32", "default": 0 },
  780. { "name": "end", "type": "int32", "default": 0 }
  781. ],
  782. "outputs": [
  783. { "name": "outputs", "option": "variadic" }
  784. ]
  785. },
  786. {
  787. "name": "SoftMax",
  788. "category": "Activation",
  789. "attributes": [
  790. { "name": "axis", "type": "int32", "default": 0 }
  791. ]
  792. },
  793. {
  794. "name": "SpaceToBatchND",
  795. "category": "Shape",
  796. "attributes": [
  797. { "name": "dilation_x", "type": "int32", "default": 0 },
  798. { "name": "dilation_y", "type": "int32", "default": 0 },
  799. { "name": "pad_top", "type": "int32", "default": 0 },
  800. { "name": "pad_bottom", "type": "int32", "default": 0 },
  801. { "name": "pad_left", "type": "int32", "default": 0 },
  802. { "name": "pad_right", "type": "int32", "default": 0 }
  803. ]
  804. },
  805. {
  806. "name": "SpaceToDepth",
  807. "category": "Shape",
  808. "attributes": [
  809. { "name": "block_size", "type": "int32", "default": 0 }
  810. ]
  811. },
  812. {
  813. "name": "SparseToDense",
  814. "category": "Shape",
  815. "attributes": [
  816. { "name": "output_shape_size0", "type": "int32", "default": 0 },
  817. { "name": "output_shape_size1", "type": "int32", "default": 0 },
  818. { "name": "default_value", "type": "int32", "default": 0 }
  819. ]
  820. },
  821. {
  822. "name": "Split",
  823. "category": "Shape",
  824. "attributes": [
  825. { "name": "axis", "type": "int32", "default": 0 },
  826. { "name": "split_dim", "type": "int32", "default": 0 },
  827. { "name": "is_caffe", "type": "boolean", "default": false },
  828. { "name": "is_onnx", "type": "boolean", "default": false },
  829. { "name": "split_sizes", "type": "int32[]", "default": [] }
  830. ]
  831. },
  832. {
  833. "name": "SquaredDifference",
  834. "category": "Layer"
  835. },
  836. {
  837. "name": "Squeeze",
  838. "category": "Transform",
  839. "attributes": [
  840. { "name": "dim_0", "type": "int32", "default": 0 },
  841. { "name": "dim_1", "type": "int32", "default": 0 },
  842. { "name": "dim_2", "type": "int32", "default": 0 },
  843. { "name": "dim_3", "type": "int32", "default": 0 }
  844. ]
  845. },
  846. {
  847. "name": "StridedSlice",
  848. "category": "Shape",
  849. "attributes": [
  850. { "name": "begine_n", "type": "int32", "default": 0 },
  851. { "name": "end_n", "type": "int32", "default": 0 },
  852. { "name": "stride_n", "type": "int32", "default": 0 },
  853. { "name": "begine_c", "type": "int32", "default": 0 },
  854. { "name": "end_c", "type": "int32", "default": 0 },
  855. { "name": "stride_c", "type": "int32", "default": 0 },
  856. { "name": "begine_h", "type": "int32", "default": 0 },
  857. { "name": "end_h", "type": "int32", "default": 0 },
  858. { "name": "stride_h", "type": "int32", "default": 0 },
  859. { "name": "begine_w", "type": "int32", "default": 0 },
  860. { "name": "end_w", "type": "int32", "default": 0 },
  861. { "name": "stride_w", "type": "int32", "default": 0 }
  862. ],
  863. "inputs": [
  864. { "name": "input" },
  865. { "name": "begin" },
  866. { "name": "end" },
  867. { "name": "strides" }
  868. ]
  869. },
  870. {
  871. "name": "SwapAxis",
  872. "category": "Shape",
  873. "attributes": [
  874. { "name": "dim_0", "type": "int32", "default": 0 },
  875. { "name": "dim_1", "type": "int32", "default": 0 }
  876. ]
  877. },
  878. {
  879. "name": "TanH",
  880. "category": "Activation"
  881. },
  882. {
  883. "name": "Threshold",
  884. "category": "Activation",
  885. "attributes": [
  886. { "name": "Threshold", "type": "float32", "default": 0 }
  887. ]
  888. },
  889. {
  890. "name": "TopKV2",
  891. "category": "Layer",
  892. "attributes": [
  893. { "name": "k", "type": "int32", "default": 0 },
  894. { "name": "sorted", "type": "int32", "default": 0 }
  895. ]
  896. },
  897. {
  898. "name": "Transpose",
  899. "category": "Transform",
  900. "attributes": [
  901. { "name": "shape", "type": "int32[]", "default": [] }
  902. ]
  903. },
  904. {
  905. "name": "Unary",
  906. "attributes": [
  907. { "name": "type", "type": "int32", "default": 0 }
  908. ]
  909. },
  910. {
  911. "name": "Unsqueeze",
  912. "category": "Transform",
  913. "attributes": [
  914. { "name": "axises[]", "type": "int32[]", "default": [] }
  915. ]
  916. },
  917. {
  918. "name": "Upsample",
  919. "category": "Data",
  920. "attributes": [
  921. { "name": "scale", "type": "float32", "default": 0 }
  922. ]
  923. },
  924. {
  925. "name": "Where",
  926. "category": "Layer"
  927. },
  928. {
  929. "name": "ZerosLike",
  930. "category": "Layer"
  931. }
  932. ]