netron/source/torch-metadata.json

[
  {
    "name": "cudnn.BatchBRNNReLU",
    "category": "Normalization"
  },
  {
    "name": "cudnn.BatchNormalization",
    "category": "Normalization",
    "attributes": [
      { "name": "affine", "default": true },
      { "name": "momentum", "default": 0.1 },
      { "name": "eps", "default": 0.00001 }
    ]
  },
  {
    "name": "cudnn.BLSTM",
    "category": "Layer"
  },
  {
    "name": "cudnn.LogSoftMax",
    "category": "Activation"
  },
  {
    "name": "cudnn.ReLU",
    "category": "Activation",
    "attributes": [
      { "name": "threshold", "default": 0 },
      { "name": "val", "default": 0 },
      { "name": "inplace", "default": false, "visible": false },
      { "name": "mode", "default": "CUDNN_ACTIVATION_RELU" },
      { "name": "nElem", "visible": false }
    ]
  },
  {
    "name": "cudnn.RNN",
    "category": "Layer"
  },
  {
    "name": "cudnn.SpatialAveragePooling",
    "category": "Pool",
    "attributes": [
      { "name": "ceil_mode", "default": false },
      { "name": "mode", "default": "CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING" },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "count_include_pad", "visible": false }
    ]
  },
  {
    "name": "cudnn.SpatialBatchNormalization",
    "category": "Normalization",
    "attributes": [
      { "name": "affine", "default": true },
      { "name": "momentum", "default": 0.1 },
      { "name": "eps", "default": 0.00001 },
      { "name": "mode", "default": "CUDNN_BATCHNORM_SPATIAL" },
      { "name": "nDim", "default": 4 },
      { "name": "__shareGradInputKey", "visible": false }
    ]
  },
  {
    "name": "cudnn.SpatialConvolution",
    "category": "Layer",
    "attributes": [
      { "name": "benchmarked", "visible": false },
      { "name": "input_offset", "visible": false },
      { "name": "output_offset", "visible": false },
      { "name": "weight_offset", "visible": false },
      { "name": "groups", "default": 1 },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "padding", "default": 0 },
      { "name": "nInputPlane", "visible": false },
      { "name": "nOutputPlane", "visible": false },
      { "name": "convDescData", "visible": false },
      { "name": "autotunerHash", "visible": false },
      { "name": "fmode", "visible": false },
      { "name": "bwmode", "visible": false },
      { "name": "bdmode", "visible": false }
    ]
  },
  {
    "name": "cudnn.SpatialFullConvolution",
    "category": "Layer",
    "attributes": [
      { "name": "groups", "default": 1 },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "dilation", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "convDescData", "visible": false },
      { "name": "autotunerHash", "visible": false },
      { "name": "nInputPlane", "visible": false },
      { "name": "nOutputPlane", "visible": false },
      { "name": "input_offset", "visible": false },
      { "name": "output_offset", "visible": false },
      { "name": "weight_offset", "visible": false }
    ]
  },
  {
    "name": "cudnn.SpatialMaxPooling",
    "category": "Pool",
    "attributes": [
      { "name": "ceil_mode", "default": false },
      { "name": "mode", "default": "CUDNN_POOLING_MAX" },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "iheight", "visible": false },
      { "name": "iwidth", "visible": false }
    ]
  },
  {
    "name": "inn.SpatialMaxPooling",
    "category": "Pool",
    "attributes": [
      { "name": "ceil_mode", "default": false },
      { "name": "mode", "default": "CUDNN_POOLING_MAX" },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "iheight", "visible": false },
      { "name": "iwidth", "visible": false }
    ]
  },
  {
    "name": "nn.BatchNormalization",
    "category": "Normalization",
    "attributes": [
      { "name": "affine", "default": true },
      { "name": "momentum", "default": 0.1 },
      { "name": "eps", "default": 0.00001 }
    ]
  },
  {
    "name": "nn.CAddTable"
  },
  {
    "name": "nn.Concat",
    "category": "Tensor",
    "attributes": [
      { "name": "outputSize", "visible": false }
    ]
  },
  {
    "name": "nn.ConcatTable"
  },
  {
    "name": "nn.Contiguous"
  },
  {
    "name": "nn.Dropout",
    "category": "Dropout",
    "attributes": [
      { "name": "v2", "visible": false }
    ]
  },
  {
    "name": "nn.FlattenTable"
  },
  {
    "name": "nn.Identity"
  },
  {
    "name": "nn.InstanceNormalization",
    "category": "Normalization",
    "attributes": [
      { "name": "affine", "default": true },
      { "name": "nOutput", "visible": false },
      { "name": "prev_batch_size", "visible": false },
      { "name": "eps", "default": 0.00001 },
      { "name": "momentum", "default": 0.1 }
    ]
  },
  {
    "name": "nn.LeakyReLU",
    "category": "Activation",
    "attributes": [
      { "name": "negval", "default": 0.01, "visible": false },
      { "name": "inplace", "default": false, "visible": false }
    ]
  },
  {
    "name": "nn.Linear",
    "category": "Layer"
  },
  {
    "name": "nn.LinearNoBias",
    "category": "Layer"
  },
  {
    "name": "nn.LogSoftMax",
    "category": "Activation"
  },
  {
    "name": "nn.Mean",
    "attributes": [
      { "name": "squeeze", "default": true },
      { "name": "sizeAverage", "default": false },
      { "name": "dimension", "default": 1 },
      { "name": "nInputDims", "visible": false }
    ]
  },
  {
    "name": "nn.MM"
  },
  {
    "name": "nn.MulConstant",
    "attributes": [
      { "name": "inplace", "default": false, "visible": false }
    ]
  },
  {
    "name": "nn.Normalize",
    "category": "Normalization",
    "attributes": []
  },
  {
    "name": "nn.Normalize2",
    "category": "Normalization",
    "attributes": []
  },
  {
    "name": "nn.PReLU",
    "category": "Activation"
  },
  {
    "name": "nn.ReLU",
    "category": "Activation",
    "attributes": [
      { "name": "threshold", "default": 0 },
      { "name": "val", "default": 0 },
      { "name": "inplace", "default": false, "visible": false },
      { "name": "mode", "default": "CUDNN_ACTIVATION_RELU" },
      { "name": "nElem", "visible": false }
    ]
  },
  {
    "name": "nn.Reshape",
    "category": "Shape",
    "attributes": [
      { "name": "nelement", "visible": false }
    ]
  },
  {
    "name": "nn.ScaleTable"
  },
  {
    "name": "nn.SelectTable"
  },
  {
    "name": "nn.Sequencer"
  },
  {
    "name": "nn.ShaveImage"
  },
  {
    "name": "nn.Sigmoid",
    "category": "Activation",
    "attributes": [
      { "name": "mode", "default": "CUDNN_ACTIVATION_SIGMOID" },
      { "name": "nElem", "visible": false }
    ]
  },
  {
    "name": "nn.SoftMax",
    "category": "Activation"
  },
  {
    "name": "nn.SpatialAveragePooling",
    "category": "Pool",
    "attributes": [
      { "name": "ceil_mode", "default": false },
      { "name": "mode", "default": "CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING" },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "count_include_pad", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialBatchNormalization",
    "category": "Normalization",
    "attributes": [
      { "name": "affine", "default": true },
      { "name": "momentum", "default": 0.1 },
      { "name": "eps", "default": 0.00001 },
      { "name": "mode", "default": "CUDNN_BATCHNORM_SPATIAL" },
      { "name": "nDim", "default": 4 },
      { "name": "__shareGradInputKey", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialConvolution",
    "category": "Layer",
    "attributes": [
      { "name": "benchmarked", "visible": false },
      { "name": "input_offset", "visible": false },
      { "name": "output_offset", "visible": false },
      { "name": "weight_offset", "visible": false },
      { "name": "groups", "default": 1 },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "padding", "default": 0 },
      { "name": "nInputPlane", "visible": false },
      { "name": "nOutputPlane", "visible": false },
      { "name": "convDescData", "visible": false },
      { "name": "autotunerHash", "visible": false },
      { "name": "fmode", "visible": false },
      { "name": "bwmode", "visible": false },
      { "name": "bdmode", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialConvolutionMM",
    "category": "Layer",
    "attributes": [
      { "name": "groups", "default": 1 },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "padding", "default": 0 },
      { "name": "nInputPlane", "visible": false },
      { "name": "nOutputPlane", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialCrossMapLRN",
    "category": "Normalization",
    "attributes": []
  },
  {
    "name": "nn.SpatialDilatedConvolution",
    "category": "Layer",
    "attributes": [
      { "name": "d", "default": [   1,   1 ] },
      { "name": "dilation", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "nInputPlane", "visible": false },
      { "name": "nOutputPlane", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialDropout",
    "category": "Dropout"
  },
  {
    "name": "nn.SpatialFractionalMaxPooling",
    "category": "Pool"
  },
  {
    "name": "nn.SpatialFullConvolution",
    "category": "Layer",
    "attributes": [
      { "name": "groups", "default": 1 },
      { "name": "d", "default": [   1,   1 ] },
      { "name": "dilation", "default": [   1,   1 ] },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "convDescData", "visible": false },
      { "name": "autotunerHash", "visible": false },
      { "name": "nInputPlane", "visible": false },
      { "name": "nOutputPlane", "visible": false },
      { "name": "input_offset", "visible": false },
      { "name": "output_offset", "visible": false },
      { "name": "weight_offset", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialMaxPooling",
    "category": "Pool",
    "attributes": [
      { "name": "ceil_mode", "default": false },
      { "name": "mode", "default": "CUDNN_POOLING_MAX" },
      { "name": "pad", "default": [   0,   0 ] },
      { "name": "iheight", "visible": false },
      { "name": "iwidth", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialMaxUnpooling",
    "category": "Pool",
    "attributes": [
      { "name": "oheight", "visible": false },
      { "name": "owidth", "visible": false }
    ]
  },
  {
    "name": "nn.SpatialReflectionPadding",
    "category": "Tensor",
    "attributes": []
  },
  {
    "name": "nn.SpatialReplicationPadding",
    "category": "Tensor",
    "attributes": []
  },
  {
    "name": "nn.SpatialSubtractiveNormalization",
    "category": "Normalization"
  },
  {
    "name": "nn.SpatialZeroPadding",
    "category": "Tensor",
    "attributes": []
  },
  {
    "name": "nn.SplitTable"
  },
  {
    "name": "nn.Squeeze",
    "category": "Transform"
  },
  {
    "name": "nn.Tanh",
    "category": "Activation",
    "attributes": [
      { "name": "mode", "default": "CUDNN_ACTIVATION_TANH" },
      { "name": "nElem", "visible": false }
    ]
  },
  {
    "name": "nn.TotalVariation"
  },
  {
    "name": "nn.Transpose",
    "category": "Transform"
  },
  {
    "name": "nn.View"
  },
  {
    "name": "nn.VolumetricAveragePooling",
    "category": "Pool"
  },
  {
    "name": "nn.VolumetricBatchNormalization",
    "category": "Normalization",
    "attributes": [
      { "name": "affine", "default": true },
      { "name": "momentum", "default": 0.1 },
      { "name": "eps", "default": 0.00001 }
    ]
  },
  {
    "name": "nn.VolumetricConvolution",
    "category": "Layer"
  },
  {
    "name": "nn.VolumetricMaxPooling",
    "category": "Pool",
    "attributes": [
      { "name": "ceil_mode", "default": false }
    ]
  },
  {
    "name": "w2nn.ScaleTable"
  }
]