| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570 |
- [
- {
- "name": "_copy",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_linalg_gemm2",
- "category": "Layer",
- "inputs": [
- { "name": "input" },
- { "name": "weights" },
- { "name": "bias" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_minus_scalar",
- "attributes": [
- { "name": "scalar", "type": "float32" }
- ],
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_mul",
- "inputs": [
- { "name": "x" },
- { "name": "y" }
- ],
- "outputs": [
- { "name": "z" }
- ]
- },
- {
- "name": "_mul_scalar",
- "attributes": [
- { "name": "scalar", "type": "float32" }
- ],
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_Plus",
- "inputs": [
- { "name": "inputs", "type": "Tensor[]" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_plus_scalar",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_rminus_scalar",
- "attributes": [
- { "name": "scalar", "type": "float32" }
- ],
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "_sub",
- "inputs": [
- { "name": "x" },
- { "name": "y" }
- ],
- "outputs": [
- { "name": "z" }
- ]
- },
- {
- "name": "_zeros",
- "category": "Constant",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Activation",
- "category": "Activation",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "BatchNorm",
- "category": "Normalization",
- "attributes": [
- { "type": "int32", "default": 1, "name": "axis" },
- { "type": "float64", "default": 0.001, "name": "eps" },
- { "type": "float32", "default": 0.9, "name": "momentum" },
- { "type": "boolean", "default": true, "name": "fix_gamma" },
- { "type": "boolean", "default": false, "name": "use_global_stats" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "gamma" },
- { "name": "beta" },
- { "name": "mean" },
- { "name": "variance" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "broadcast_add",
- "inputs": [
- { "name": "lhs" },
- { "name": "rhs" }
- ],
- "outputs": [
- { "name": "out" }
- ]
- },
- {
- "name": "broadcast_div",
- "inputs": [
- { "name": "lhs" },
- { "name": "rhs" }
- ],
- "outputs": [
- { "name": "out" }
- ]
- },
- {
- "name": "broadcast_mul",
- "inputs": [
- { "name": "lhs" },
- { "name": "rhs" }
- ],
- "outputs": [
- { "name": "out" }
- ]
- },
- {
- "name": "Concat",
- "category": "Tensor",
- "attributes": [
- { "default": "1", "name": "dim" },
- { "visible": false, "name": "num_args" }
- ],
- "inputs": [
- { "name": "inputs", "type": "Tensor[]" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Convolution",
- "category": "Layer",
- "attributes": [
- { "default": false, "name": "cudnn_off", "type": "boolean" },
- { "default": "off", "name": "cudnn_tune" },
- { "default": [ 1, null ], "name": "dilate", "type": "int32[]" },
- { "name": "kernel", "type": "int32[]" },
- { "visible": false, "name": "no_bias", "type": "boolean" },
- { "type": "int32", "default": 1, "name": "num_group" },
- { "type": "int32", "name": "num_filter" },
- { "default": [ 0, null ], "name": "pad", "type": "int32[]" },
- { "default": [ 1, null ], "name": "stride", "type": "int32[]" },
- { "type": "int32", "default": "1024", "name": "workspace" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "weight" },
- { "name": "bias", "optional": true }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "CuDNNBatchNorm",
- "category": "Normalization",
- "inputs": [
- { "name": "input" },
- { "name": "gamma" },
- { "name": "beta" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Deconvolution",
- "category": "Layer",
- "attributes": [
- { "visible": false, "name": "no_bias" },
- { "default": "1", "name": "num_group" },
- { "type": "int32", "default": "1024", "name": "workspace" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "weight" },
- { "name": "bias" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Dropout",
- "category": "Dropout",
- "attributes": [
- { "type": "float32", "default": 0.5, "name": "p" }
- ],
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "ElementWiseSum",
- "category": "Normalization",
- "inputs": [
- { "name": "inputs", "type": "Tensor[]" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "elemwise_add",
- "inputs": [
- { "name": "lhs" },
- { "name": "rhs" }
- ],
- "outputs": [
- { "name": "out" }
- ]
- },
- {
- "name": "elemwise_div",
- "inputs": [
- { "name": "lhs" },
- { "name": "rhs" }
- ],
- "outputs": [
- { "name": "out" }
- ]
- },
- {
- "name": "elemwise_sub",
- "inputs": [
- { "name": "lhs" },
- { "name": "rhs" }
- ],
- "outputs": [
- { "name": "out" }
- ]
- },
- {
- "name": "Embedding",
- "category": "Transform",
- "attributes": [
- { "type": "int32", "name": "input_dim" },
- { "type": "int32", "name": "output_dim" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "weight" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Flatten",
- "category": "Shape",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "FullyConnected",
- "category": "Layer",
- "attributes": [
- { "type": "boolean", "default": true, "name": "flatten" },
- { "type": "boolean", "visible": false, "name": "no_bias" },
- { "type": "int32", "name": "num_hidden" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "weight" },
- { "name": "bias" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "LayerNorm",
- "category": "Normalization",
- "attributes": [
- { "type": "int32", "default": 1, "name": "axis" },
- { "type": "float64", "default": 0.001, "name": "eps" },
- { "type": "float32", "default": 0.9, "name": "momentum" },
- { "type": "boolean", "default": true, "name": "fix_gamma" },
- { "type": "boolean", "default": false, "name": "use_global_stats" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "gamma" },
- { "name": "beta" },
- { "name": "mean" },
- { "name": "variance" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "LeakyReLU",
- "category": "Activation",
- "inputs": [
- { "name": "input" },
- { "name": "weight" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "log",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "LRN",
- "category": "Normalization",
- "attributes": [
- { "name": "alpha", "type": "float32", "default": 0.0001 },
- { "name": "beta", "type": "float32", "default": 0.75 },
- { "name": "knorm", "type": "float32", "default": 2 },
- { "name": "nsize", "type": "int32" }
- ],
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "MakeLoss",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "mean",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Pad",
- "category": "Tensor",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Pooling",
- "category": "Pool",
- "attributes": [
- { "default": false, "name": "cudnn_off" },
- { "default": false, "name": "global_pool" },
- { "name": "kernel", "type": "int32[]" },
- { "default": [ 0, null ], "name": "pad", "type": "int32[]" },
- { "default": "valid", "name": "pooling_convention" },
- { "default": "max", "name": "pool_type" },
- { "default": [ 1, null ], "name": "stride", "type": "int32[]" }
- ],
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "relu",
- "category": "Activation",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "Reshape",
- "category": "Shape",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "RNN",
- "category": "Layer",
- "attributes": [
- { "type": "boolean", "name": "bidirectional", "default": false },
- { "name": "lstm_parameters", "visible": false },
- { "type": "int32", "name": "num_layers" },
- { "type": "boolean", "default": false, "name": "state_outputs" },
- { "type": "int32", "name": "state_size" },
- { "type": "float32", "name": "p", "default": 0 }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "state_0" },
- { "name": "state_1" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "slice_axis",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "SliceChannel",
- "inputs": [
- { "name": "inputs" }
- ],
- "outputs": [
- { "name": "outputs", "type": "Tensor[]" }
- ]
- },
- {
- "name": "softmax",
- "category": "Activation",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "SoftmaxActivation",
- "category": "Activation",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "SoftmaxOutput",
- "category": "Activation",
- "attributes": [
- { "default": "1", "name": "grad_scale" },
- { "default": "-1", "name": "ignore_label" },
- { "default": false, "name": "multi_output" },
- { "default": "null", "name": "normalization" },
- { "default": false, "name": "out_grad" },
- { "default": "0", "name": "smooth_alpha" },
- { "default": false, "name": "use_ignore" },
- { "default": false, "name": "preserve_shape" }
- ],
- "inputs": [
- { "name": "input" },
- { "name": "label" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "sqrt",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "square",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "sum",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- },
- {
- "name": "transpose",
- "category": "Transform",
- "inputs": [
- { "name": "input" }
- ],
- "outputs": [
- { "name": "output" }
- ]
- }
- ]
|