uff-metadata.json 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. [
  2. {
  3. "name": "_FlattenConcat_TRT",
  4. "category": "Tensor",
  5. "inputs": [
  6. { "name": "inputs", "list": true }
  7. ]
  8. },
  9. {
  10. "name": "_GridAnchor_TRT",
  11. "inputs": [
  12. { "name": "input" }
  13. ]
  14. },
  15. {
  16. "name": "_MaxPool",
  17. "category": "Pool",
  18. "inputs": [
  19. { "name": "input" }
  20. ]
  21. },
  22. {
  23. "name": "_NMS_TRT",
  24. "inputs": [
  25. { "name": "input" },
  26. { "name": "?" },
  27. { "name": "?" }
  28. ]
  29. },
  30. {
  31. "name": "Activation",
  32. "category": "Activation",
  33. "inputs": [
  34. { "name": "input" }
  35. ]
  36. },
  37. {
  38. "name": "BatchNorm",
  39. "category": "Normalization",
  40. "inputs": [
  41. { "name": "input" },
  42. { "name": "gamma" },
  43. { "name": "beta" },
  44. { "name": "moving_mean" },
  45. { "name": "moving_variance" }
  46. ]
  47. },
  48. {
  49. "name": "Binary",
  50. "inputs": [
  51. { "name": "x" },
  52. { "name": "y" }
  53. ]
  54. },
  55. {
  56. "name": "Concat",
  57. "category": "Tensor",
  58. "inputs": [
  59. { "name": "inputs", "list": true }
  60. ]
  61. },
  62. {
  63. "name": "Conv",
  64. "category": "Layer",
  65. "inputs": [
  66. { "name": "input" },
  67. { "name": "kernel" }
  68. ]
  69. },
  70. {
  71. "name": "Flatten",
  72. "category": "Shape"
  73. },
  74. {
  75. "name": "FullyConnected",
  76. "category": "Layer",
  77. "inputs": [
  78. { "name": "input" },
  79. { "name": "weights" }
  80. ]
  81. },
  82. {
  83. "name": "GatherV2",
  84. "category": "Data",
  85. "inputs": [
  86. { "name": "input" },
  87. { "name": "indices" }
  88. ]
  89. },
  90. {
  91. "name": "Pool",
  92. "category": "Pool",
  93. "inputs": [
  94. { "name": "input" }
  95. ]
  96. },
  97. {
  98. "name": "Reshape",
  99. "category": "Shape",
  100. "inputs": [
  101. { "name": "input" },
  102. { "name": "shape" }
  103. ]
  104. },
  105. {
  106. "name": "Shape",
  107. "inputs": [
  108. { "name": "input" }
  109. ]
  110. },
  111. {
  112. "name": "Softmax",
  113. "category": "Activation"
  114. },
  115. {
  116. "name": "Squeeze",
  117. "category": "Transform"
  118. },
  119. {
  120. "name": "Stack",
  121. "inputs": [
  122. { "name": "input" },
  123. { "name": "?" },
  124. { "name": "?" },
  125. { "name": "?" }
  126. ]
  127. },
  128. {
  129. "name": "StridedSlice",
  130. "category": "Tensor",
  131. "inputs": [
  132. { "name": "input" },
  133. { "name": "begin" },
  134. { "name": "end" },
  135. { "name": "strides" }
  136. ]
  137. },
  138. {
  139. "name": "Unary",
  140. "inputs": [
  141. { "name": "input" }
  142. ]
  143. }
  144. ]