bigdl-metadata.json 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. [
  2. {
  3. "name": "com.intel.analytics.bigdl.nn.Dropout",
  4. "category": "Dropout"
  5. },
  6. {
  7. "name": "com.intel.analytics.bigdl.nn.InferReshape",
  8. "category": "Shape"
  9. },
  10. {
  11. "name": "com.intel.analytics.bigdl.nn.JoinTable",
  12. "category": "Tensor",
  13. "inputs": [
  14. { "name": "inputs", "list": true }
  15. ],
  16. "outputs": [
  17. { "name": "output" }
  18. ]
  19. },
  20. {
  21. "name": "com.intel.analytics.bigdl.nn.Linear",
  22. "category": "Layer",
  23. "inputs": [
  24. { "name": "inputs" },
  25. { "name": "weight" },
  26. { "name": "bias" }
  27. ]
  28. },
  29. {
  30. "name": "com.intel.analytics.bigdl.nn.NormalizeScale",
  31. "category": "Normalization",
  32. "inputs": [
  33. { "name": "inputs" },
  34. { "name": "w" }
  35. ]
  36. },
  37. {
  38. "name": "com.intel.analytics.bigdl.nn.ReLU",
  39. "category": "Activation"
  40. },
  41. {
  42. "name": "Scale",
  43. "category": "Layer",
  44. "inputs": [
  45. { "name": "inputs" },
  46. { "name": "weight" },
  47. { "name": "bias" }
  48. ]
  49. },
  50. {
  51. "name": "SoftMax",
  52. "category": "Activation"
  53. },
  54. {
  55. "name": "com.intel.analytics.bigdl.nn.SpatialAveragePooling",
  56. "category": "Pool"
  57. },
  58. {
  59. "name": "com.intel.analytics.bigdl.nn.SpatialBatchNormalization",
  60. "category": "Normalization"
  61. },
  62. {
  63. "name": "com.intel.analytics.bigdl.nn.quantized.SpatialConvolution",
  64. "category": "Layer",
  65. "inputs": [
  66. { "name": "inputs" },
  67. { "name": "weight" },
  68. { "name": "bias" }
  69. ]
  70. },
  71. {
  72. "name": "com.intel.analytics.bigdl.nn.SpatialCrossMapLRN",
  73. "category": "Normalization"
  74. },
  75. {
  76. "name": "com.intel.analytics.bigdl.nn.SpatialDilatedConvolution",
  77. "category": "Layer",
  78. "inputs": [
  79. { "name": "inputs" },
  80. { "name": "weight" },
  81. { "name": "bias" }
  82. ]
  83. },
  84. {
  85. "name": "com.intel.analytics.bigdl.nn.SpatialMaxPooling",
  86. "category": "Pool"
  87. },
  88. {
  89. "name": "com.intel.analytics.bigdl.nn.Transpose",
  90. "category": "Shape"
  91. },
  92. {
  93. "name": "com.intel.analytics.bigdl.nn.View"
  94. }
  95. ]