2
0

onednn-metadata.json 6.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412
  1. [
  2. {
  3. "name": "End"
  4. },
  5. {
  6. "name": "Wildcard"
  7. },
  8. {
  9. "name": "StaticReshape",
  10. "category": "Shape"
  11. },
  12. {
  13. "name": "AvgPoolBackward",
  14. "category": "Pool"
  15. },
  16. {
  17. "name": "AvgPool",
  18. "category": "Pool"
  19. },
  20. {
  21. "name": "MaxPoolBackward",
  22. "category": "Pool"
  23. },
  24. {
  25. "name": "MaxPool",
  26. "category": "Pool"
  27. },
  28. {
  29. "name": "LayerNormBackward",
  30. "category": "Normalization",
  31. "inputs": [
  32. { "name": "input" },
  33. { "name": "diff_dst" },
  34. { "name": "mean" },
  35. { "name": "variance" },
  36. { "name": "gamma" },
  37. { "name": "beta" }
  38. ],
  39. "outputs": [
  40. { "name": "diff_src" },
  41. { "name": "diff_gamma" },
  42. { "name": "diff_beta" }
  43. ]
  44. },
  45. {
  46. "name": "BatchNormInference",
  47. "category": "Normalization",
  48. "inputs": [
  49. { "name": "input" },
  50. { "name": "gamma" },
  51. { "name": "beta" },
  52. { "name": "mean" },
  53. { "name": "variance" }
  54. ],
  55. "outputs": [
  56. { "name": "output" }
  57. ]
  58. },
  59. {
  60. "name": "BatchNormTrainingBackward",
  61. "category": "Normalization",
  62. "inputs": [
  63. { "name": "input" },
  64. { "name": "diff_dst" },
  65. { "name": "mean" },
  66. { "name": "variance" },
  67. { "name": "gamma" }
  68. ],
  69. "outputs": [
  70. { "name": "diff_src" },
  71. { "name": "diff_gamma" },
  72. { "name": "diff_beta" }
  73. ]
  74. },
  75. {
  76. "name": "BatchNormForwardTraining",
  77. "category": "Normalization",
  78. "inputs": [
  79. { "name": "input" },
  80. { "name": "mean" },
  81. { "name": "variance" },
  82. { "name": "gamma" },
  83. { "name": "beta" }
  84. ],
  85. "outputs": [
  86. { "name": "output" },
  87. { "name": "running_mean" },
  88. { "name": "running_variance" },
  89. { "name": "batch_mean" },
  90. { "name": "batch_variance" }
  91. ]
  92. },
  93. {
  94. "name": "LayerNorm",
  95. "category": "Normalization",
  96. "inputs": [
  97. { "name": "input" },
  98. { "name": "gamma" },
  99. { "name": "beta" }
  100. ],
  101. "outputs": [
  102. { "name": "dst" },
  103. { "name": "mean" },
  104. { "name": "variance" }
  105. ]
  106. },
  107. {
  108. "name": "DynamicDequantize",
  109. "category": "Quantization"
  110. },
  111. {
  112. "name": "DynamicQuantize",
  113. "category": "Quantization"
  114. },
  115. {
  116. "name": "Dequantize",
  117. "category": "Quantization"
  118. },
  119. {
  120. "name": "Quantize",
  121. "category": "Quantization"
  122. },
  123. {
  124. "name": "TypeCast"
  125. },
  126. {
  127. "name": "ReLU",
  128. "category": "Activation"
  129. },
  130. {
  131. "name": "Sigmoid",
  132. "category": "Activation"
  133. },
  134. {
  135. "name": "SoftMaxBackward",
  136. "category": "Activation"
  137. },
  138. {
  139. "name": "MishBackward",
  140. "category": "Activation"
  141. },
  142. {
  143. "name": "GELU",
  144. "category": "Activation"
  145. },
  146. {
  147. "name": "ClampBackward",
  148. "category": "Activation"
  149. },
  150. {
  151. "name": "Mish",
  152. "category": "Activation"
  153. },
  154. {
  155. "name": "PReLU",
  156. "category": "Activation"
  157. },
  158. {
  159. "name": "SigmoidBackward",
  160. "category": "Activation"
  161. },
  162. {
  163. "name": "LogSoftmax",
  164. "category": "Activation"
  165. },
  166. {
  167. "name": "Clamp",
  168. "category": "Activation"
  169. },
  170. {
  171. "name": "TanhBackward",
  172. "category": "Activation"
  173. },
  174. {
  175. "name": "ReLUBackward",
  176. "category": "Activation"
  177. },
  178. {
  179. "name": "Exp",
  180. "category": "Activation"
  181. },
  182. {
  183. "name": "SoftMax",
  184. "category": "Activation"
  185. },
  186. {
  187. "name": "HardSwish",
  188. "category": "Activation"
  189. },
  190. {
  191. "name": "LeakyReLU",
  192. "category": "Activation"
  193. },
  194. {
  195. "name": "LogSoftmaxBackward",
  196. "category": "Activation"
  197. },
  198. {
  199. "name": "Tanh",
  200. "category": "Activation"
  201. },
  202. {
  203. "name": "Elu",
  204. "category": "Activation"
  205. },
  206. {
  207. "name": "SoftPlus",
  208. "category": "Activation"
  209. },
  210. {
  211. "name": "EluBackward",
  212. "category": "Activation"
  213. },
  214. {
  215. "name": "GELUBackward",
  216. "category": "Activation"
  217. },
  218. {
  219. "name": "SoftPlusBackward",
  220. "category": "Activation"
  221. },
  222. {
  223. "name": "PReLUBackward",
  224. "category": "Activation"
  225. },
  226. {
  227. "name": "HardSwishBackward",
  228. "category": "Activation"
  229. },
  230. {
  231. "name": "Interpolate"
  232. },
  233. {
  234. "name": "InterpolateBackward"
  235. },
  236. {
  237. "name": "ReduceMax"
  238. },
  239. {
  240. "name": "ReduceL2"
  241. },
  242. {
  243. "name": "ReduceL1"
  244. },
  245. {
  246. "name": "ReduceSum"
  247. },
  248. {
  249. "name": "ReduceProd"
  250. },
  251. {
  252. "name": "ReduceMean"
  253. },
  254. {
  255. "name": "ReduceMin"
  256. },
  257. {
  258. "name": "Reorder",
  259. "category": "Transform"
  260. },
  261. {
  262. "name": "Concat",
  263. "category": "Tensor"
  264. },
  265. {
  266. "name": "StaticTranspose",
  267. "category": "Transform"
  268. },
  269. {
  270. "name": "Abs"
  271. },
  272. {
  273. "name": "Sqrt"
  274. },
  275. {
  276. "name": "Round"
  277. },
  278. {
  279. "name": "Subtract"
  280. },
  281. {
  282. "name": "Erf"
  283. },
  284. {
  285. "name": "Minimum"
  286. },
  287. {
  288. "name": "Reciprocal"
  289. },
  290. {
  291. "name": "BiasAddBackward"
  292. },
  293. {
  294. "name": "Divide"
  295. },
  296. {
  297. "name": "SquaredDifference"
  298. },
  299. {
  300. "name": "Multiply"
  301. },
  302. {
  303. "name": "AbsBackward"
  304. },
  305. {
  306. "name": "Square"
  307. },
  308. {
  309. "name": "Maximum"
  310. },
  311. {
  312. "name": "Add"
  313. },
  314. {
  315. "name": "SqrtBackward"
  316. },
  317. {
  318. "name": "Log"
  319. },
  320. {
  321. "name": "BiasAdd",
  322. "inputs": [
  323. { "name": "input" },
  324. { "name": "bias" }
  325. ],
  326. "outputs": [
  327. { "name": "output" }
  328. ]
  329. },
  330. {
  331. "name": "ConvTranspose",
  332. "category": "Layer",
  333. "inputs": [
  334. { "name": "input" },
  335. { "name": "weights" },
  336. { "name": "bias" }
  337. ],
  338. "outputs": [
  339. { "name": "output" }
  340. ]
  341. },
  342. {
  343. "name": "Convolution",
  344. "category": "Layer",
  345. "inputs": [
  346. { "name": "input" },
  347. { "name": "weights" },
  348. { "name": "bias" }
  349. ],
  350. "outputs": [
  351. { "name": "output" }
  352. ]
  353. },
  354. {
  355. "name": "ConvTransposeBackwardData",
  356. "category": "Layer",
  357. "inputs": [
  358. { "name": "input" },
  359. { "name": "weights" }
  360. ],
  361. "outputs": [
  362. { "name": "output" }
  363. ]
  364. },
  365. {
  366. "name": "ConvolutionBackwardData",
  367. "category": "Layer",
  368. "inputs": [
  369. { "name": "input" },
  370. { "name": "weights" },
  371. { "name": "dst_shape" }
  372. ],
  373. "outputs": [
  374. { "name": "output" }
  375. ]
  376. },
  377. {
  378. "name": "ConvolutionBackwardWeights",
  379. "category": "Layer",
  380. "inputs": [
  381. { "name": "input" },
  382. { "name": "diff_dst" },
  383. { "name": "weights_shape" }
  384. ],
  385. "outputs": [
  386. { "name": "output" }
  387. ]
  388. },
  389. {
  390. "name": "ConvTransposeBackwardWeights",
  391. "category": "Layer",
  392. "inputs": [
  393. { "name": "input" },
  394. { "name": "diff_dst" },
  395. { "name": "weights_shape" }
  396. ],
  397. "outputs": [
  398. { "name": "output" }
  399. ]
  400. },
  401. {
  402. "name": "MatMul",
  403. "inputs": [
  404. { "name": "input" },
  405. { "name": "weights" },
  406. { "name": "bias" }
  407. ],
  408. "outputs": [
  409. { "name": "output" }
  410. ]
  411. }
  412. ]