[ { "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" } ]