rknn-metadata.json 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313
  1. [
  2. {
  3. "name": "VSI_NN_OP_ADD",
  4. "inputs": [
  5. { "name": "input" },
  6. { "name": "other" }
  7. ]
  8. },
  9. {
  10. "name": "VSI_NN_OP_CONCAT",
  11. "category": "Tensor",
  12. "inputs": [
  13. { "name": "inputs", "list": true }
  14. ]
  15. },
  16. {
  17. "name": "VSI_NN_OP_CONV_RELU",
  18. "category": "Layer",
  19. "inputs": [
  20. { "name": "input" },
  21. { "name": "weights" },
  22. { "name": "bias" }
  23. ]
  24. },
  25. {
  26. "name": "VSI_NN_OP_CONV_RELU_POOL",
  27. "category": "Layer",
  28. "inputs": [
  29. { "name": "input" },
  30. { "name": "weights" },
  31. { "name": "bias" }
  32. ]
  33. },
  34. {
  35. "name": "VSI_NN_OP_CONV2D",
  36. "category": "Layer",
  37. "inputs": [
  38. { "name": "input" },
  39. { "name": "weights" },
  40. { "name": "bias" }
  41. ]
  42. },
  43. {
  44. "name": "VSI_NN_OP_DECONVOLUTION",
  45. "category": "Layer",
  46. "inputs": [
  47. { "name": "input" },
  48. { "name": "weights" },
  49. { "name": "bias" }
  50. ]
  51. },
  52. {
  53. "name": "VSI_NN_OP_FCL",
  54. "category": "Layer",
  55. "inputs": [
  56. { "name": "input" },
  57. { "name": "weights" },
  58. { "name": "bias" }
  59. ]
  60. },
  61. {
  62. "name": "VSI_NN_OP_FCL_RELU",
  63. "category": "Layer",
  64. "inputs": [
  65. { "name": "input" },
  66. { "name": "weights" },
  67. { "name": "bias" }
  68. ]
  69. },
  70. {
  71. "name": "VSI_NN_OP_LEAKY_RELU",
  72. "category": "Activation"
  73. },
  74. {
  75. "name": "VSI_NN_OP_LEAKY_SIGMOID",
  76. "category": "Activation"
  77. },
  78. {
  79. "name": "VSI_NN_OP_MISH",
  80. "category": "Activation"
  81. },
  82. {
  83. "name": "VSI_NN_OP_PERMUTE",
  84. "category": "Shape"
  85. },
  86. {
  87. "name": "VSI_NN_OP_POOL",
  88. "category": "Pool"
  89. },
  90. {
  91. "name": "VSI_NN_OP_RELU",
  92. "category": "Activation"
  93. },
  94. {
  95. "name": "VSI_NN_OP_PRELU",
  96. "category": "Activation",
  97. "inputs": [
  98. { "name": "input" },
  99. { "name": "slope" }
  100. ]
  101. },
  102. {
  103. "name": "VSI_NN_OP_RESHAPE",
  104. "category": "Shape"
  105. },
  106. {
  107. "name": "VSI_NN_OP_SIGMOID",
  108. "category": "Activation"
  109. },
  110. {
  111. "name": "VSI_NN_OP_SOFTMAX",
  112. "category": "Activation"
  113. },
  114. {
  115. "name": "VSI_NN_OP_RESIZE",
  116. "category": "Shape"
  117. },
  118. {
  119. "name": "VSI_NN_OP_LRN",
  120. "category": "Normalization"
  121. },
  122. {
  123. "name": "VSI_NN_OP_BATCH_NORM",
  124. "category": "Normalization",
  125. "inputs": [
  126. { "name": "input" },
  127. { "name": "gamma" },
  128. { "name": "beta" },
  129. { "name": "mean" },
  130. { "name": "variance" }
  131. ]
  132. },
  133. {
  134. "name": "VSI_NN_OP_LSTM",
  135. "category": "Layer"
  136. },
  137. {
  138. "name": "VSI_NN_OP_STRIDED_SLICE",
  139. "category": "Tensor"
  140. },
  141. {
  142. "name": "VSI_NN_OP_SWISH",
  143. "category": "Activation"
  144. },
  145. {
  146. "name": "VSI_NN_OP_RELUN",
  147. "category": "Activation"
  148. },
  149. {
  150. "name": "VSI_NN_OP_INSTANCE_NORM",
  151. "category": "Normalization",
  152. "inputs": [
  153. { "name": "input" },
  154. { "name": "scale" },
  155. { "name": "bias" }
  156. ]
  157. },
  158. {
  159. "name": "Conv",
  160. "category": "Layer",
  161. "inputs": [
  162. { "name": "input" },
  163. { "name": "weights" },
  164. { "name": "bias" }
  165. ]
  166. },
  167. {
  168. "name": "ConvTranspose",
  169. "category": "Layer",
  170. "inputs": [
  171. { "name": "input" },
  172. { "name": "weights" },
  173. { "name": "bias" }
  174. ]
  175. },
  176. {
  177. "name": "ConvRelu",
  178. "category": "Layer",
  179. "inputs": [
  180. { "name": "input" },
  181. { "name": "weights" },
  182. { "name": "bias" }
  183. ]
  184. },
  185. {
  186. "name": "ConvClip",
  187. "category": "Layer",
  188. "inputs": [
  189. { "name": "input" },
  190. { "name": "weights" },
  191. { "name": "bias" }
  192. ]
  193. },
  194. {
  195. "name": "Concat",
  196. "category": "Tensor",
  197. "inputs": [
  198. { "name": "inputs", "list": true }
  199. ]
  200. },
  201. {
  202. "name": "BatchNormalization",
  203. "category": "Normalization",
  204. "inputs": [
  205. { "name": "input" },
  206. { "name": "weight" },
  207. { "name": "bias" },
  208. { "name": "running_mean" },
  209. { "name": "running_var" }
  210. ]
  211. },
  212. {
  213. "name": "Relu",
  214. "category": "Activation"
  215. },
  216. {
  217. "name": "Softmax",
  218. "category": "Activation"
  219. },
  220. {
  221. "name": "MaxPool",
  222. "category": "Pool"
  223. },
  224. {
  225. "name": "AveragePool",
  226. "category": "Pool"
  227. },
  228. {
  229. "name": "Reshape",
  230. "category": "Shape",
  231. "inputs": [
  232. { "name": "input" },
  233. { "name": "shape" }
  234. ]
  235. },
  236. {
  237. "name": "Transpose",
  238. "category": "Transform"
  239. },
  240. {
  241. "name": "Add",
  242. "inputs": [
  243. { "name": "A" },
  244. { "name": "B" }
  245. ],
  246. "outputs": [
  247. { "name": "C" }
  248. ]
  249. },
  250. {
  251. "name": "Split",
  252. "category": "Tensor",
  253. "outputs": [
  254. { "name": "output", "list": true }
  255. ]
  256. },
  257. {
  258. "name": "PoolingLayer2",
  259. "category": "Pool"
  260. },
  261. {
  262. "name": "Softmax2Layer",
  263. "category": "Activation"
  264. },
  265. {
  266. "name": "SoftMax2",
  267. "category": "Activation"
  268. },
  269. {
  270. "name": "TensorTranspose",
  271. "category": "Transform"
  272. },
  273. {
  274. "name": "ActivationLayer",
  275. "category": "Activation"
  276. },
  277. {
  278. "name": "LeakyReluLayer",
  279. "category": "Activation"
  280. },
  281. {
  282. "name": "ConvolutionLayer",
  283. "category": "Layer"
  284. },
  285. {
  286. "name": "TensorScale",
  287. "category": "Layer"
  288. },
  289. {
  290. "name": "BatchNormalizationLayer",
  291. "category": "Normalization"
  292. },
  293. {
  294. "name": "Sigmoid",
  295. "category": "Activation"
  296. },
  297. {
  298. "name": "LeakyRelu",
  299. "category": "Activation"
  300. },
  301. {
  302. "name": "HardSigmoid",
  303. "category": "Activation"
  304. },
  305. {
  306. "name": "exSwish",
  307. "category": "Activation"
  308. },
  309. {
  310. "name": "Slice",
  311. "category": "Tensor"
  312. }
  313. ]