onednn-metadata.json 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414
  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. "category": "Quantization"
  126. },
  127. {
  128. "name": "ReLU",
  129. "category": "Activation"
  130. },
  131. {
  132. "name": "Sigmoid",
  133. "category": "Activation"
  134. },
  135. {
  136. "name": "SoftMaxBackward",
  137. "category": "Activation"
  138. },
  139. {
  140. "name": "MishBackward",
  141. "category": "Activation"
  142. },
  143. {
  144. "name": "GELU",
  145. "category": "Activation"
  146. },
  147. {
  148. "name": "ClampBackward",
  149. "category": "Activation"
  150. },
  151. {
  152. "name": "Mish",
  153. "category": "Activation"
  154. },
  155. {
  156. "name": "PReLU",
  157. "category": "Activation"
  158. },
  159. {
  160. "name": "SigmoidBackward",
  161. "category": "Activation"
  162. },
  163. {
  164. "name": "LogSoftmax",
  165. "category": "Activation"
  166. },
  167. {
  168. "name": "Clamp",
  169. "category": "Activation"
  170. },
  171. {
  172. "name": "TanhBackward",
  173. "category": "Activation"
  174. },
  175. {
  176. "name": "ReLUBackward",
  177. "category": "Activation"
  178. },
  179. {
  180. "name": "Exp",
  181. "category": "Activation"
  182. },
  183. {
  184. "name": "SoftMax",
  185. "category": "Activation"
  186. },
  187. {
  188. "name": "HardSwish",
  189. "category": "Activation"
  190. },
  191. {
  192. "name": "LeakyReLU",
  193. "category": "Activation"
  194. },
  195. {
  196. "name": "LogSoftmaxBackward",
  197. "category": "Activation"
  198. },
  199. {
  200. "name": "Tanh",
  201. "category": "Activation"
  202. },
  203. {
  204. "name": "Elu",
  205. "category": "Activation"
  206. },
  207. {
  208. "name": "SoftPlus",
  209. "category": "Activation"
  210. },
  211. {
  212. "name": "EluBackward",
  213. "category": "Activation"
  214. },
  215. {
  216. "name": "GELUBackward",
  217. "category": "Activation"
  218. },
  219. {
  220. "name": "SoftPlusBackward",
  221. "category": "Activation"
  222. },
  223. {
  224. "name": "PReLUBackward",
  225. "category": "Activation"
  226. },
  227. {
  228. "name": "HardSwishBackward",
  229. "category": "Activation"
  230. },
  231. {
  232. "name": "Interpolate"
  233. },
  234. {
  235. "name": "InterpolateBackward"
  236. },
  237. {
  238. "name": "ReduceMax"
  239. },
  240. {
  241. "name": "ReduceL2"
  242. },
  243. {
  244. "name": "ReduceL1"
  245. },
  246. {
  247. "name": "ReduceSum"
  248. },
  249. {
  250. "name": "ReduceProd"
  251. },
  252. {
  253. "name": "ReduceMean"
  254. },
  255. {
  256. "name": "ReduceMin"
  257. },
  258. {
  259. "name": "Reorder",
  260. "category": "Transform"
  261. },
  262. {
  263. "name": "Concat",
  264. "category": "Tensor"
  265. },
  266. {
  267. "name": "StaticTranspose",
  268. "category": "Transform"
  269. },
  270. {
  271. "name": "Abs"
  272. },
  273. {
  274. "name": "Sqrt"
  275. },
  276. {
  277. "name": "Round"
  278. },
  279. {
  280. "name": "Subtract"
  281. },
  282. {
  283. "name": "Erf"
  284. },
  285. {
  286. "name": "Minimum"
  287. },
  288. {
  289. "name": "Reciprocal"
  290. },
  291. {
  292. "name": "BiasAddBackward"
  293. },
  294. {
  295. "name": "Divide"
  296. },
  297. {
  298. "name": "SquaredDifference"
  299. },
  300. {
  301. "name": "Multiply"
  302. },
  303. {
  304. "name": "AbsBackward"
  305. },
  306. {
  307. "name": "Square"
  308. },
  309. {
  310. "name": "Maximum"
  311. },
  312. {
  313. "name": "Add"
  314. },
  315. {
  316. "name": "SqrtBackward"
  317. },
  318. {
  319. "name": "Log"
  320. },
  321. {
  322. "name": "BiasAdd",
  323. "inputs": [
  324. { "name": "input" },
  325. { "name": "bias" }
  326. ],
  327. "outputs": [
  328. { "name": "output" }
  329. ]
  330. },
  331. {
  332. "name": "ConvTranspose",
  333. "category": "Layer",
  334. "inputs": [
  335. { "name": "input" },
  336. { "name": "weights" },
  337. { "name": "bias" }
  338. ],
  339. "outputs": [
  340. { "name": "output" }
  341. ]
  342. },
  343. {
  344. "name": "Convolution",
  345. "category": "Layer",
  346. "inputs": [
  347. { "name": "input" },
  348. { "name": "weights" },
  349. { "name": "bias" }
  350. ],
  351. "outputs": [
  352. { "name": "output" }
  353. ]
  354. },
  355. {
  356. "name": "ConvTransposeBackwardData",
  357. "category": "Layer",
  358. "inputs": [
  359. { "name": "input" },
  360. { "name": "weights" }
  361. ],
  362. "outputs": [
  363. { "name": "output" }
  364. ]
  365. },
  366. {
  367. "name": "ConvolutionBackwardData",
  368. "category": "Layer",
  369. "inputs": [
  370. { "name": "input" },
  371. { "name": "weights" },
  372. { "name": "dst_shape" }
  373. ],
  374. "outputs": [
  375. { "name": "output" }
  376. ]
  377. },
  378. {
  379. "name": "ConvolutionBackwardWeights",
  380. "category": "Layer",
  381. "inputs": [
  382. { "name": "input" },
  383. { "name": "diff_dst" },
  384. { "name": "weights_shape" }
  385. ],
  386. "outputs": [
  387. { "name": "output" }
  388. ]
  389. },
  390. {
  391. "name": "ConvTransposeBackwardWeights",
  392. "category": "Layer",
  393. "inputs": [
  394. { "name": "input" },
  395. { "name": "diff_dst" },
  396. { "name": "weights_shape" }
  397. ],
  398. "outputs": [
  399. { "name": "output" }
  400. ]
  401. },
  402. {
  403. "name": "MatMul",
  404. "category": "Layer",
  405. "inputs": [
  406. { "name": "input" },
  407. { "name": "weights" },
  408. { "name": "bias" }
  409. ],
  410. "outputs": [
  411. { "name": "output" }
  412. ]
  413. }
  414. ]