mxnet-metadata.json 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570
  1. [
  2. {
  3. "name": "_copy",
  4. "inputs": [
  5. { "name": "input" }
  6. ],
  7. "outputs": [
  8. { "name": "output" }
  9. ]
  10. },
  11. {
  12. "name": "_linalg_gemm2",
  13. "category": "Layer",
  14. "inputs": [
  15. { "name": "input" },
  16. { "name": "weights" },
  17. { "name": "bias" }
  18. ],
  19. "outputs": [
  20. { "name": "output" }
  21. ]
  22. },
  23. {
  24. "name": "_minus_scalar",
  25. "attributes": [
  26. { "name": "scalar", "type": "float32" }
  27. ],
  28. "inputs": [
  29. { "name": "input" }
  30. ],
  31. "outputs": [
  32. { "name": "output" }
  33. ]
  34. },
  35. {
  36. "name": "_mul",
  37. "inputs": [
  38. { "name": "x" },
  39. { "name": "y" }
  40. ],
  41. "outputs": [
  42. { "name": "z" }
  43. ]
  44. },
  45. {
  46. "name": "_mul_scalar",
  47. "attributes": [
  48. { "name": "scalar", "type": "float32" }
  49. ],
  50. "inputs": [
  51. { "name": "input" }
  52. ],
  53. "outputs": [
  54. { "name": "output" }
  55. ]
  56. },
  57. {
  58. "name": "_Plus",
  59. "inputs": [
  60. { "name": "inputs", "type": "Tensor[]" }
  61. ],
  62. "outputs": [
  63. { "name": "output" }
  64. ]
  65. },
  66. {
  67. "name": "_plus_scalar",
  68. "inputs": [
  69. { "name": "input" }
  70. ],
  71. "outputs": [
  72. { "name": "output" }
  73. ]
  74. },
  75. {
  76. "name": "_rminus_scalar",
  77. "attributes": [
  78. { "name": "scalar", "type": "float32" }
  79. ],
  80. "inputs": [
  81. { "name": "input" }
  82. ],
  83. "outputs": [
  84. { "name": "output" }
  85. ]
  86. },
  87. {
  88. "name": "_sub",
  89. "inputs": [
  90. { "name": "x" },
  91. { "name": "y" }
  92. ],
  93. "outputs": [
  94. { "name": "z" }
  95. ]
  96. },
  97. {
  98. "name": "_zeros",
  99. "category": "Constant",
  100. "inputs": [
  101. { "name": "input" }
  102. ],
  103. "outputs": [
  104. { "name": "output" }
  105. ]
  106. },
  107. {
  108. "name": "Activation",
  109. "category": "Activation",
  110. "inputs": [
  111. { "name": "input" }
  112. ],
  113. "outputs": [
  114. { "name": "output" }
  115. ]
  116. },
  117. {
  118. "name": "BatchNorm",
  119. "category": "Normalization",
  120. "attributes": [
  121. { "type": "int32", "default": 1, "name": "axis" },
  122. { "type": "float64", "default": 0.001, "name": "eps" },
  123. { "type": "float32", "default": 0.9, "name": "momentum" },
  124. { "type": "boolean", "default": true, "name": "fix_gamma" },
  125. { "type": "boolean", "default": false, "name": "use_global_stats" }
  126. ],
  127. "inputs": [
  128. { "name": "input" },
  129. { "name": "gamma" },
  130. { "name": "beta" },
  131. { "name": "mean" },
  132. { "name": "variance" }
  133. ],
  134. "outputs": [
  135. { "name": "output" }
  136. ]
  137. },
  138. {
  139. "name": "broadcast_add",
  140. "inputs": [
  141. { "name": "lhs" },
  142. { "name": "rhs" }
  143. ],
  144. "outputs": [
  145. { "name": "out" }
  146. ]
  147. },
  148. {
  149. "name": "broadcast_div",
  150. "inputs": [
  151. { "name": "lhs" },
  152. { "name": "rhs" }
  153. ],
  154. "outputs": [
  155. { "name": "out" }
  156. ]
  157. },
  158. {
  159. "name": "broadcast_mul",
  160. "inputs": [
  161. { "name": "lhs" },
  162. { "name": "rhs" }
  163. ],
  164. "outputs": [
  165. { "name": "out" }
  166. ]
  167. },
  168. {
  169. "name": "Concat",
  170. "category": "Tensor",
  171. "attributes": [
  172. { "default": "1", "name": "dim" },
  173. { "visible": false, "name": "num_args" }
  174. ],
  175. "inputs": [
  176. { "name": "inputs", "type": "Tensor[]" }
  177. ],
  178. "outputs": [
  179. { "name": "output" }
  180. ]
  181. },
  182. {
  183. "name": "Convolution",
  184. "category": "Layer",
  185. "attributes": [
  186. { "default": false, "name": "cudnn_off", "type": "boolean" },
  187. { "default": "off", "name": "cudnn_tune" },
  188. { "default": [ 1, null ], "name": "dilate", "type": "int32[]" },
  189. { "name": "kernel", "type": "int32[]" },
  190. { "visible": false, "name": "no_bias", "type": "boolean" },
  191. { "type": "int32", "default": 1, "name": "num_group" },
  192. { "type": "int32", "name": "num_filter" },
  193. { "default": [ 0, null ], "name": "pad", "type": "int32[]" },
  194. { "default": [ 1, null ], "name": "stride", "type": "int32[]" },
  195. { "type": "int32", "default": "1024", "name": "workspace" }
  196. ],
  197. "inputs": [
  198. { "name": "input" },
  199. { "name": "weight" },
  200. { "name": "bias", "optional": true }
  201. ],
  202. "outputs": [
  203. { "name": "output" }
  204. ]
  205. },
  206. {
  207. "name": "CuDNNBatchNorm",
  208. "category": "Normalization",
  209. "inputs": [
  210. { "name": "input" },
  211. { "name": "gamma" },
  212. { "name": "beta" }
  213. ],
  214. "outputs": [
  215. { "name": "output" }
  216. ]
  217. },
  218. {
  219. "name": "Deconvolution",
  220. "category": "Layer",
  221. "attributes": [
  222. { "visible": false, "name": "no_bias" },
  223. { "default": "1", "name": "num_group" },
  224. { "type": "int32", "default": "1024", "name": "workspace" }
  225. ],
  226. "inputs": [
  227. { "name": "input" },
  228. { "name": "weight" },
  229. { "name": "bias" }
  230. ],
  231. "outputs": [
  232. { "name": "output" }
  233. ]
  234. },
  235. {
  236. "name": "Dropout",
  237. "category": "Dropout",
  238. "attributes": [
  239. { "type": "float32", "default": 0.5, "name": "p" }
  240. ],
  241. "inputs": [
  242. { "name": "input" }
  243. ],
  244. "outputs": [
  245. { "name": "output" }
  246. ]
  247. },
  248. {
  249. "name": "ElementWiseSum",
  250. "category": "Normalization",
  251. "inputs": [
  252. { "name": "inputs", "type": "Tensor[]" }
  253. ],
  254. "outputs": [
  255. { "name": "output" }
  256. ]
  257. },
  258. {
  259. "name": "elemwise_add",
  260. "inputs": [
  261. { "name": "lhs" },
  262. { "name": "rhs" }
  263. ],
  264. "outputs": [
  265. { "name": "out" }
  266. ]
  267. },
  268. {
  269. "name": "elemwise_div",
  270. "inputs": [
  271. { "name": "lhs" },
  272. { "name": "rhs" }
  273. ],
  274. "outputs": [
  275. { "name": "out" }
  276. ]
  277. },
  278. {
  279. "name": "elemwise_sub",
  280. "inputs": [
  281. { "name": "lhs" },
  282. { "name": "rhs" }
  283. ],
  284. "outputs": [
  285. { "name": "out" }
  286. ]
  287. },
  288. {
  289. "name": "Embedding",
  290. "category": "Transform",
  291. "attributes": [
  292. { "type": "int32", "name": "input_dim" },
  293. { "type": "int32", "name": "output_dim" }
  294. ],
  295. "inputs": [
  296. { "name": "input" },
  297. { "name": "weight" }
  298. ],
  299. "outputs": [
  300. { "name": "output" }
  301. ]
  302. },
  303. {
  304. "name": "Flatten",
  305. "category": "Shape",
  306. "inputs": [
  307. { "name": "input" }
  308. ],
  309. "outputs": [
  310. { "name": "output" }
  311. ]
  312. },
  313. {
  314. "name": "FullyConnected",
  315. "category": "Layer",
  316. "attributes": [
  317. { "type": "boolean", "default": true, "name": "flatten" },
  318. { "type": "boolean", "visible": false, "name": "no_bias" },
  319. { "type": "int32", "name": "num_hidden" }
  320. ],
  321. "inputs": [
  322. { "name": "input" },
  323. { "name": "weight" },
  324. { "name": "bias" }
  325. ],
  326. "outputs": [
  327. { "name": "output" }
  328. ]
  329. },
  330. {
  331. "name": "LayerNorm",
  332. "category": "Normalization",
  333. "attributes": [
  334. { "type": "int32", "default": 1, "name": "axis" },
  335. { "type": "float64", "default": 0.001, "name": "eps" },
  336. { "type": "float32", "default": 0.9, "name": "momentum" },
  337. { "type": "boolean", "default": true, "name": "fix_gamma" },
  338. { "type": "boolean", "default": false, "name": "use_global_stats" }
  339. ],
  340. "inputs": [
  341. { "name": "input" },
  342. { "name": "gamma" },
  343. { "name": "beta" },
  344. { "name": "mean" },
  345. { "name": "variance" }
  346. ],
  347. "outputs": [
  348. { "name": "output" }
  349. ]
  350. },
  351. {
  352. "name": "LeakyReLU",
  353. "category": "Activation",
  354. "inputs": [
  355. { "name": "input" },
  356. { "name": "weight" }
  357. ],
  358. "outputs": [
  359. { "name": "output" }
  360. ]
  361. },
  362. {
  363. "name": "log",
  364. "inputs": [
  365. { "name": "input" }
  366. ],
  367. "outputs": [
  368. { "name": "output" }
  369. ]
  370. },
  371. {
  372. "name": "LRN",
  373. "category": "Normalization",
  374. "attributes": [
  375. { "name": "alpha", "type": "float32", "default": 0.0001 },
  376. { "name": "beta", "type": "float32", "default": 0.75 },
  377. { "name": "knorm", "type": "float32", "default": 2 },
  378. { "name": "nsize", "type": "int32" }
  379. ],
  380. "inputs": [
  381. { "name": "input" }
  382. ],
  383. "outputs": [
  384. { "name": "output" }
  385. ]
  386. },
  387. {
  388. "name": "MakeLoss",
  389. "inputs": [
  390. { "name": "input" }
  391. ],
  392. "outputs": [
  393. { "name": "output" }
  394. ]
  395. },
  396. {
  397. "name": "mean",
  398. "inputs": [
  399. { "name": "input" }
  400. ],
  401. "outputs": [
  402. { "name": "output" }
  403. ]
  404. },
  405. {
  406. "name": "Pad",
  407. "category": "Tensor",
  408. "inputs": [
  409. { "name": "input" }
  410. ],
  411. "outputs": [
  412. { "name": "output" }
  413. ]
  414. },
  415. {
  416. "name": "Pooling",
  417. "category": "Pool",
  418. "attributes": [
  419. { "default": false, "name": "cudnn_off" },
  420. { "default": false, "name": "global_pool" },
  421. { "name": "kernel", "type": "int32[]" },
  422. { "default": [ 0, null ], "name": "pad", "type": "int32[]" },
  423. { "default": "valid", "name": "pooling_convention" },
  424. { "default": "max", "name": "pool_type" },
  425. { "default": [ 1, null ], "name": "stride", "type": "int32[]" }
  426. ],
  427. "inputs": [
  428. { "name": "input" }
  429. ],
  430. "outputs": [
  431. { "name": "output" }
  432. ]
  433. },
  434. {
  435. "name": "relu",
  436. "category": "Activation",
  437. "inputs": [
  438. { "name": "input" }
  439. ],
  440. "outputs": [
  441. { "name": "output" }
  442. ]
  443. },
  444. {
  445. "name": "Reshape",
  446. "category": "Shape",
  447. "inputs": [
  448. { "name": "input" }
  449. ],
  450. "outputs": [
  451. { "name": "output" }
  452. ]
  453. },
  454. {
  455. "name": "RNN",
  456. "category": "Layer",
  457. "attributes": [
  458. { "type": "boolean", "name": "bidirectional", "default": false },
  459. { "name": "lstm_parameters", "visible": false },
  460. { "type": "int32", "name": "num_layers" },
  461. { "type": "boolean", "default": false, "name": "state_outputs" },
  462. { "type": "int32", "name": "state_size" },
  463. { "type": "float32", "name": "p", "default": 0 }
  464. ],
  465. "inputs": [
  466. { "name": "input" },
  467. { "name": "state_0" },
  468. { "name": "state_1" }
  469. ],
  470. "outputs": [
  471. { "name": "output" }
  472. ]
  473. },
  474. {
  475. "name": "slice_axis",
  476. "inputs": [
  477. { "name": "input" }
  478. ],
  479. "outputs": [
  480. { "name": "output" }
  481. ]
  482. },
  483. {
  484. "name": "SliceChannel",
  485. "inputs": [
  486. { "name": "inputs" }
  487. ],
  488. "outputs": [
  489. { "name": "outputs", "type": "Tensor[]" }
  490. ]
  491. },
  492. {
  493. "name": "softmax",
  494. "category": "Activation",
  495. "inputs": [
  496. { "name": "input" }
  497. ],
  498. "outputs": [
  499. { "name": "output" }
  500. ]
  501. },
  502. {
  503. "name": "SoftmaxActivation",
  504. "category": "Activation",
  505. "inputs": [
  506. { "name": "input" }
  507. ],
  508. "outputs": [
  509. { "name": "output" }
  510. ]
  511. },
  512. {
  513. "name": "SoftmaxOutput",
  514. "category": "Activation",
  515. "attributes": [
  516. { "default": "1", "name": "grad_scale" },
  517. { "default": "-1", "name": "ignore_label" },
  518. { "default": false, "name": "multi_output" },
  519. { "default": "null", "name": "normalization" },
  520. { "default": false, "name": "out_grad" },
  521. { "default": "0", "name": "smooth_alpha" },
  522. { "default": false, "name": "use_ignore" },
  523. { "default": false, "name": "preserve_shape" }
  524. ],
  525. "inputs": [
  526. { "name": "input" },
  527. { "name": "label" }
  528. ],
  529. "outputs": [
  530. { "name": "output" }
  531. ]
  532. },
  533. {
  534. "name": "sqrt",
  535. "inputs": [
  536. { "name": "input" }
  537. ],
  538. "outputs": [
  539. { "name": "output" }
  540. ]
  541. },
  542. {
  543. "name": "square",
  544. "inputs": [
  545. { "name": "input" }
  546. ],
  547. "outputs": [
  548. { "name": "output" }
  549. ]
  550. },
  551. {
  552. "name": "sum",
  553. "inputs": [
  554. { "name": "input" }
  555. ],
  556. "outputs": [
  557. { "name": "output" }
  558. ]
  559. },
  560. {
  561. "name": "transpose",
  562. "category": "Transform",
  563. "inputs": [
  564. { "name": "input" }
  565. ],
  566. "outputs": [
  567. { "name": "output" }
  568. ]
  569. }
  570. ]