tengine-metadata.json 27 KB

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