wehub-resource-sync
7254f7b4d1
Build / Build (macos-latest) (push) Has been cancelled
Build / Build (ubuntu-latest) (push) Has been cancelled
Build / Build (windows-latest) (push) Has been cancelled
Build / Analyze (javascript) (push) Has been cancelled
Build / Analyze (python) (push) Has been cancelled
465 lines
12 KiB
JSON
465 lines
12 KiB
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.normal3DConv",
|
|
"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": "cudnn.VolumetricBatchNormalization",
|
|
"category": "Normalization",
|
|
"attributes": [
|
|
{ "name": "affine", "default": true },
|
|
{ "name": "momentum", "default": 0.1 },
|
|
{ "name": "eps", "default": 0.00001 }
|
|
]
|
|
},
|
|
{
|
|
"name": "cudnn.VolumetricMaxPooling",
|
|
"category": "Pool",
|
|
"attributes": [
|
|
{ "name": "ceil_mode", "default": 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"
|
|
},
|
|
{
|
|
"name": "nn.Normalize2",
|
|
"category": "Normalization"
|
|
},
|
|
{
|
|
"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"
|
|
},
|
|
{
|
|
"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"
|
|
},
|
|
{
|
|
"name": "nn.SpatialReplicationPadding",
|
|
"category": "Tensor"
|
|
},
|
|
{
|
|
"name": "nn.SpatialSubtractiveNormalization",
|
|
"category": "Normalization"
|
|
},
|
|
{
|
|
"name": "nn.SpatialZeroPadding",
|
|
"category": "Tensor"
|
|
},
|
|
{
|
|
"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"
|
|
}
|
|
] |