paddle-metadata.json 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. [
  2. {
  3. "name": "batch_norm",
  4. "category": "Normalization",
  5. "attributes": [
  6. { "name": "momentum", "default": 0.8999999761581421 },
  7. { "name": "epsilon", "default": 0.000009999999747378752 },
  8. { "name": "fuse_with_relu", "default": false },
  9. { "name": "data_layout", "default": "NCHW" }
  10. ]
  11. },
  12. {
  13. "name": "concat",
  14. "category": "Tensor"
  15. },
  16. {
  17. "name": "conv2d",
  18. "category": "Layer",
  19. "attributes": [
  20. { "name": "workspace_size_MB", "default": 4096 },
  21. { "name": "fuse_residual_connection", "default": false },
  22. { "name": "fuse_eltwise", "default": false },
  23. { "name": "fuse_relu", "default": false },
  24. { "name": "data_format", "default": "AnyLayout" },
  25. { "name": "groups", "default": 1 },
  26. { "name": "paddings", "default": [ 0, 0 ] },
  27. { "name": "dilations", "default": [ 1, 1 ] },
  28. { "name": "strides", "default": [ 1, 1 ] }
  29. ]
  30. },
  31. {
  32. "name": "conv2d_transpose",
  33. "category": "Layer"
  34. },
  35. {
  36. "name": "depthwise_conv2d",
  37. "category": "Layer",
  38. "attributes": [
  39. { "name": "workspace_size_MB", "default": 4096 },
  40. { "name": "fuse_residual_connection", "default": false },
  41. { "name": "data_format", "default": "AnyLayout" },
  42. { "name": "groups", "default": 1 },
  43. { "name": "fuse_relu", "default": false }
  44. ]
  45. },
  46. {
  47. "name": "elementwise_add",
  48. "attributes": [
  49. { "name": "axis", "default": -1 }
  50. ]
  51. },
  52. {
  53. "name": "lrn",
  54. "category": "Normalization",
  55. "attributes": [
  56. { "name": "alpha", "default": 0.00009999999747378752 },
  57. { "name": "beta", "default": 0.75 },
  58. { "name": "k", "default": 1 }
  59. ]
  60. },
  61. {
  62. "name": "pad2d",
  63. "category": "Tensor"
  64. },
  65. {
  66. "name": "pool2d",
  67. "category": "Pool",
  68. "attributes": [
  69. { "name": "data_format", "default": "AnyLayout" },
  70. { "name": "ceil_mode", "default": false },
  71. { "name": "global_pooling", "default": false },
  72. { "name": "exclusive", "default": true },
  73. { "name": "pooling_type", "default": "max" },
  74. { "name": "paddings", "default": [ 0, 0 ] }
  75. ]
  76. },
  77. {
  78. "name": "hard_swish",
  79. "category": "Activation"
  80. },
  81. {
  82. "name": "hard_sigmoid",
  83. "category": "Activation"
  84. },
  85. {
  86. "name": "relu",
  87. "category": "Activation"
  88. },
  89. {
  90. "name": "sigmoid",
  91. "category": "Activation"
  92. },
  93. {
  94. "name": "reshape",
  95. "category": "Shape"
  96. },
  97. {
  98. "name": "reshape2",
  99. "category": "Shape"
  100. },
  101. {
  102. "name": "scale",
  103. "category": "Layer"
  104. },
  105. {
  106. "name": "rnn",
  107. "category": "Layer"
  108. },
  109. {
  110. "name": "transpose",
  111. "category": "Transform"
  112. },
  113. {
  114. "name": "transpose2",
  115. "category": "Transform"
  116. },
  117. {
  118. "name": "softmax",
  119. "category": "Activation",
  120. "attributes": [
  121. { "name": "data_format", "default": "AnyLayout" }
  122. ]
  123. }
  124. ]