mxnet-metadata.json 12 KB

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