[ { "name": "batch_norm", "category": "Normalization", "attributes": [ { "name": "momentum", "default": 0.8999999761581421 }, { "name": "epsilon", "default": 0.000009999999747378752 }, { "name": "fuse_with_relu", "default": false }, { "name": "data_layout", "default": "NCHW" } ] }, { "name": "concat", "category": "Tensor" }, { "name": "conv2d", "category": "Layer", "attributes": [ { "name": "workspace_size_MB", "default": 4096 }, { "name": "fuse_residual_connection", "default": false }, { "name": "fuse_eltwise", "default": false }, { "name": "fuse_relu", "default": false }, { "name": "data_format", "default": "AnyLayout" }, { "name": "groups", "default": 1 }, { "name": "paddings", "default": [ 0, 0 ] }, { "name": "dilations", "default": [ 1, 1 ] }, { "name": "strides", "default": [ 1, 1 ] } ] }, { "name": "conv2d_transpose", "category": "Layer" }, { "name": "depthwise_conv2d", "category": "Layer", "attributes": [ { "name": "workspace_size_MB", "default": 4096 }, { "name": "fuse_residual_connection", "default": false }, { "name": "data_format", "default": "AnyLayout" }, { "name": "groups", "default": 1 }, { "name": "fuse_relu", "default": false } ] }, { "name": "elementwise_add", "attributes": [ { "name": "axis", "default": -1 } ] }, { "name": "lrn", "category": "Normalization", "attributes": [ { "name": "alpha", "default": 0.00009999999747378752 }, { "name": "beta", "default": 0.75 }, { "name": "k", "default": 1 } ] }, { "name": "pad2d", "category": "Tensor" }, { "name": "pool2d", "category": "Pool", "attributes": [ { "name": "data_format", "default": "AnyLayout" }, { "name": "ceil_mode", "default": false }, { "name": "global_pooling", "default": false }, { "name": "exclusive", "default": true }, { "name": "pooling_type", "default": "max" }, { "name": "paddings", "default": [ 0, 0 ] } ] }, { "name": "hard_swish", "category": "Activation" }, { "name": "hard_sigmoid", "category": "Activation" }, { "name": "relu", "category": "Activation" }, { "name": "sigmoid", "category": "Activation" }, { "name": "reshape", "category": "Shape" }, { "name": "reshape2", "category": "Shape" }, { "name": "scale", "category": "Layer" }, { "name": "rnn", "category": "Layer" }, { "name": "transpose", "category": "Transform" }, { "name": "transpose2", "category": "Transform" }, { "name": "softmax", "category": "Activation", "attributes": [ { "name": "data_format", "default": "AnyLayout" } ] } ]