wehub/lutzroeder--netron
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity
Files
main
lutzroeder--netron/source/megengine-metadata.json
T
wehub-resource-sync 7254f7b4d1
Build / Build (macos-latest) (push) Has been cancelled
Details
Build / Build (ubuntu-latest) (push) Has been cancelled
Details
Build / Build (windows-latest) (push) Has been cancelled
Details
Build / Analyze (javascript) (push) Has been cancelled
Details
Build / Analyze (python) (push) Has been cancelled
Details
chore: import upstream snapshot with attribution
2026-07-13 12:37:45 +08:00

3532 lines
90 KiB
JSON
Raw Permalink Blame History

[
{
"name": "megengine.distributed.functional.all_gather",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" },
{ "name": "axis" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.all_reduce_max",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.all_reduce_min",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.all_reduce_sum",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.all_to_all",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" },
{ "name": "split_axis" },
{ "name": "concat_axis" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.broadcast",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.collective_comm",
"attributes": [
{ "name": "inp" },
{ "name": "mode" },
{ "name": "group" },
{ "name": "device" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.gather",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" },
{ "name": "axis" }
],
"category": "Transform"
},
{
"name": "megengine.distributed.functional.reduce_scatter_sum",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" },
{ "name": "axis" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.reduce_sum",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.remote_recv",
"attributes": [
{ "name": "src_rank" },
{ "name": "device" },
{ "name": "inp" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.remote_send",
"attributes": [
{ "name": "inp" },
{ "name": "dest_rank" }
],
"category": "Layer"
},
{
"name": "megengine.distributed.functional.scatter",
"attributes": [
{ "name": "inp" },
{ "name": "group" },
{ "name": "device" },
{ "name": "axis" }
],
"category": "Layer"
},
{
"name": "megengine.functional.elemwise.abs",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.add",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.acos",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.asin",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.atan",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.atan2",
"attributes": [
{ "name": "y" },
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.asinh",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.acosh",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.atanh",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.ceil",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.clip",
"attributes": [
{ "name": "x" },
{ "name": "lower" },
{ "name": "upper" }
]
},
{
"name": "megengine.functional.elemwise.cos",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.cosh",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.div",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.equal",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.exp",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.expm1",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.floor",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.floor_div",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.greater",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.greater_equal",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.left_shift",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.less",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.less_equal",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.log",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.log1p",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.logical_and",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.logical_not",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.logical_or",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.logical_xor",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.logaddexp",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.maximum",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.minimum",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.mod",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.mul",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.neg",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.not_equal",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.pow",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.right_shift",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.round",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.sin",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.sinh",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.sqrt",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.square",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.sub",
"attributes": [
{ "name": "x" },
{ "name": "y" }
]
},
{
"name": "megengine.functional.elemwise.tan",
"attributes": [
{ "name": "x" }
]
},
{
"name": "megengine.functional.elemwise.tanh",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.loss.l1_loss",
"category": "Layer",
"varargs": "args"
},
{
"name": "megengine.functional.loss.square_loss",
"category": "Layer",
"varargs": "args"
},
{
"name": "megengine.functional.loss.cross_entropy",
"category": "Layer",
"varargs": "args"
},
{
"name": "megengine.functional.loss.binary_cross_entropy",
"category": "Layer",
"varargs": "args"
},
{
"name": "megengine.functional.loss.hinge_loss",
"category": "Layer",
"varargs": "args"
},
{
"name": "megengine.functional.loss.ctc_loss",
"attributes": [
{ "name": "pred" },
{ "name": "pred_lengths" },
{ "name": "label" },
{ "name": "label_lengths" },
{ "name": "blank" },
{ "name": "reduction" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.argmax",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.argmin",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.argsort",
"attributes": [
{ "name": "inp" },
{ "name": "descending" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.dot",
"attributes": [
{ "name": "inp1" },
{ "name": "inp2" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.isinf",
"attributes": [
{ "name": "inp" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.isnan",
"attributes": [
{ "name": "inp" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.matinv",
"attributes": [
{ "name": "inp" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.matmul",
"attributes": [
{ "name": "inp1" },
{ "name": "inp2" },
{ "name": "transpose_a" },
{ "name": "transpose_b" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.max",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.mean",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.min",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.norm",
"attributes": [
{ "name": "inp" },
{ "name": "ord" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.normalize",
"attributes": [
{ "name": "inp" },
{ "name": "ord" },
{ "name": "axis" },
{ "name": "eps" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.prod",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.sign",
"attributes": [
{ "name": "inp" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.sort",
"attributes": [
{ "name": "inp" },
{ "name": "descending" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.std",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.sum",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.svd",
"attributes": [
{ "name": "inp" },
{ "name": "full_matrices" },
{ "name": "compute_uv" }
],
"category": "Layer"
},
{
"name": "megengine.functional.math.topk",
"category": "Layer",
"varargs": "args"
},
{
"name": "megengine.functional.math.var",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.metric.topk_accuracy",
"attributes": [
{ "name": "logits" },
{ "name": "target" },
{ "name": "topk" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.adaptive_avg_pool2d",
"attributes": [
{ "name": "inp" },
{ "name": "oshp" }
],
"category": "Pool"
},
{
"name": "megengine.functional.nn.adaptive_max_pool2d",
"attributes": [
{ "name": "inp" },
{ "name": "oshp" }
],
"category": "Pool"
},
{
"name": "megengine.functional.nn.avg_pool2d",
"attributes": [
{ "name": "inp" },
{ "name": "kernel_size" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "mode" }
],
"category": "Pool"
},
{
"name": "megengine.functional.nn.batch_norm",
"attributes": [
{ "name": "inp" },
{ "name": "running_mean" },
{ "name": "running_var" },
{ "name": "weight" },
{ "name": "bias" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.conv1d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.conv2d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.conv3d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "conv_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.conv_transpose2d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.conv_transpose3d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.deformable_conv2d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "offset" },
{ "name": "mask" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.deformable_psroi_pooling",
"attributes": [
{ "name": "inp" },
{ "name": "rois" },
{ "name": "trans" },
{ "name": "no_trans" },
{ "name": "part_size" },
{ "name": "pooled_h" },
{ "name": "pooled_w" },
{ "name": "sample_per_part" },
{ "name": "spatial_scale" },
{ "name": "trans_std" }
],
"category": "Pool"
},
{
"name": "megengine.functional.nn.dropout",
"attributes": [
{ "name": "inp" },
{ "name": "drop_prob" },
{ "name": "training" }
],
"category": "Dropout"
},
{
"name": "megengine.functional.nn.embedding",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "padding_idx" },
{ "name": "max_norm" },
{ "name": "norm_type" }
],
"category": "Transform"
},
{
"name": "megengine.functional.nn.gelu",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.hsigmoid",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.hswish",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.indexing_one_hot",
"attributes": [
{ "name": "src" },
{ "name": "index" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.layer_norm",
"attributes": [
{ "name": "inp" },
{ "name": "normalized_shape" },
{ "name": "affine" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "eps" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.leaky_relu",
"attributes": [
{ "name": "inp" },
{ "name": "negative_slope" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.linear",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.local_conv2d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "conv_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.local_response_norm",
"attributes": [
{ "name": "inp" },
{ "name": "kernel_size" },
{ "name": "k" },
{ "name": "alpha" },
{ "name": "beta" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.logsigmoid",
"attributes": [
{ "name": "inp" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.logsumexp",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "keepdims" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.logsoftmax",
"attributes": [
{ "name": "inp" },
{ "name": "axis" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.max_pool2d",
"attributes": [
{ "name": "inp" },
{ "name": "kernel_size" },
{ "name": "stride" },
{ "name": "padding" }
],
"category": "Pool"
},
{
"name": "megengine.functional.nn.one_hot",
"attributes": [
{ "name": "inp" },
{ "name": "num_classes" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.prelu",
"attributes": [
{ "name": "inp" },
{ "name": "weight" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.pad",
"attributes": [
{ "name": "src" },
{ "name": "pad_width" },
{ "name": "mode" },
{ "name": "constant_value" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.relu",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.relu6",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.vision.remap",
"attributes": [
{ "name": "inp" },
{ "name": "map_xy" },
{ "name": "border_mode" },
{ "name": "scalar" },
{ "name": "interp_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.sigmoid",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.sliding_window",
"attributes": [
{ "name": "inp" },
{ "name": "kernel_size" },
{ "name": "padding" },
{ "name": "stride" },
{ "name": "dilation" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.sliding_window_transpose",
"attributes": [
{ "name": "inp" },
{ "name": "output_size" },
{ "name": "kernel_size" },
{ "name": "padding" },
{ "name": "stride" },
{ "name": "dilation" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.silu",
"attributes": [
{ "name": "x" }
],
"category": "Activation"
},
{
"name": "megengine.functional.nn.softmax",
"attributes": [
{ "name": "inp" },
{ "name": "axis" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.softplus",
"attributes": [
{ "name": "inp" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.sync_batch_norm",
"attributes": [
{ "name": "inp" },
{ "name": "running_mean" },
{ "name": "running_var" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "training" },
{ "name": "momentum" },
{ "name": "eps" },
{ "name": "eps_mode" },
{ "name": "group" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.warp_affine",
"attributes": [
{ "name": "inp" },
{ "name": "mat" },
{ "name": "out_shape" },
{ "name": "border_mode" },
{ "name": "border_val" },
{ "name": "format" },
{ "name": "interp_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.warp_perspective",
"attributes": [
{ "name": "inp" },
{ "name": "mat" },
{ "name": "out_shape" },
{ "name": "mat_idx" },
{ "name": "border_mode" },
{ "name": "border_val" },
{ "name": "format" },
{ "name": "interp_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.nn.pixel_shuffle",
"attributes": [
{ "name": "inp" },
{ "name": "upscale_factor" }
],
"category": "Layer"
},
{
"name": "megengine.functional.debug_param.get_execution_strategy",
"category": "Layer"
},
{
"name": "megengine.functional.quantized.conv_bias_activation",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "dtype" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "nonlinear_mode" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.quantized.batch_conv_bias_activation",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "dtype" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "nonlinear_mode" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.quantized.conv_transpose2d",
"attributes": [
{ "name": "inp" },
{ "name": "weight" },
{ "name": "bias" },
{ "name": "dtype" },
{ "name": "stride" },
{ "name": "padding" },
{ "name": "dilation" },
{ "name": "groups" },
{ "name": "conv_mode" },
{ "name": "compute_mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.tensor.arange",
"attributes": [
{ "name": "start" },
{ "name": "stop" },
{ "name": "step" },
{ "name": "dtype" },
{ "name": "device" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.broadcast_to",
"attributes": [
{ "name": "inp" },
{ "name": "shape" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.concat",
"attributes": [
{ "name": "inps" },
{ "name": "axis" },
{ "name": "device" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.cond_take",
"attributes": [
{ "name": "mask" },
{ "name": "x" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.cumsum",
"attributes": [
{ "name": "inp" },
{ "name": "axis" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.diag",
"attributes": [
{ "name": "inp" },
{ "name": "k" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.expand_dims",
"attributes": [
{ "name": "inp" },
{ "name": "axis" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.eye",
"attributes": [
{ "name": "N" },
{ "name": "M" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.flatten",
"attributes": [
{ "name": "inp" },
{ "name": "start_axis" },
{ "name": "end_axis" }
],
"category": "Shape"
},
{
"name": "megengine.functional.tensor.full",
"attributes": [
{ "name": "shape" },
{ "name": "value" },
{ "name": "dtype" },
{ "name": "device" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.full_like",
"attributes": [
{ "name": "inp" },
{ "name": "value" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.gather",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "index" }
],
"category": "Transform"
},
{
"name": "megengine.functional.tensor.linspace",
"attributes": [
{ "name": "start" },
{ "name": "stop" },
{ "name": "num" },
{ "name": "dtype" },
{ "name": "device" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.ones",
"attributes": [
{ "name": "shape" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.ones_like",
"attributes": [
{ "name": "inp" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.repeat",
"attributes": [
{ "name": "inp" },
{ "name": "repeats" },
{ "name": "axis" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.reshape",
"attributes": [
{ "name": "inp" },
{ "name": "target_shape" }
],
"category": "Shape"
},
{
"name": "megengine.functional.tensor.roll",
"attributes": [
{ "name": "inp" },
{ "name": "shift" },
{ "name": "axis" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.split",
"attributes": [
{ "name": "inp" },
{ "name": "nsplits_or_sections" },
{ "name": "axis" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.squeeze",
"attributes": [
{ "name": "inp" },
{ "name": "axis" }
],
"category": "Transform"
},
{
"name": "megengine.functional.tensor.stack",
"attributes": [
{ "name": "inps" },
{ "name": "axis" },
{ "name": "device" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.scatter",
"attributes": [
{ "name": "inp" },
{ "name": "axis" },
{ "name": "index" },
{ "name": "source" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.tile",
"attributes": [
{ "name": "inp" },
{ "name": "reps" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.copy",
"attributes": [
{ "name": "inp" },
{ "name": "device" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.transpose",
"attributes": [
{ "name": "inp" },
{ "name": "pattern" }
],
"category": "Transform"
},
{
"name": "megengine.functional.tensor.swapaxes",
"attributes": [
{ "name": "inp" },
{ "name": "axis1" },
{ "name": "axis2" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.where",
"attributes": [
{ "name": "mask" },
{ "name": "x" },
{ "name": "y" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.zeros",
"attributes": [
{ "name": "shape" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.tensor.zeros_like",
"attributes": [
{ "name": "inp" }
],
"category": "Tensor"
},
{
"name": "megengine.functional.vision.correlation",
"attributes": [
{ "name": "data1" },
{ "name": "data2" },
{ "name": "kernel_size" },
{ "name": "max_displacement" },
{ "name": "stride1" },
{ "name": "stride2" },
{ "name": "pad_size" },
{ "name": "is_multiply" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.cvt_color",
"attributes": [
{ "name": "inp" },
{ "name": "mode" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.interpolate",
"attributes": [
{ "name": "inp" },
{ "name": "size" },
{ "name": "scale_factor" },
{ "name": "mode" },
{ "name": "align_corners" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.nms",
"attributes": [
{ "name": "boxes" },
{ "name": "scores" },
{ "name": "iou_thresh" },
{ "name": "max_output" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.nvof",
"attributes": [
{ "name": "src" },
{ "name": "precision" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.roi_align",
"attributes": [
{ "name": "inp" },
{ "name": "rois" },
{ "name": "output_shape" },
{ "name": "mode" },
{ "name": "spatial_scale" },
{ "name": "sample_points" },
{ "name": "aligned" }
],
"category": "Layer"
},
{
"name": "megengine.functional.vision.roi_pooling",
"attributes": [
{ "name": "inp" },
{ "name": "rois" },
{ "name": "output_shape" },
{ "name": "mode" },
{ "name": "scale" }
],
"category": "Pool"
},
{
"name": "__lt__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__le__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__gt__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ge__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__eq__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ne__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__neg__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__pos__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__abs__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__invert__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__round__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__floor__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__ceil__",
"attributes": [
{ "name": "self" }
]
},
{
"name": "__add__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__sub__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__mul__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__matmul__",
"attributes": [
{ "name": "self" },
{ "name": "other" }
]
},
{
"name": "__truediv__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__floordiv__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__mod__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__pow__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__lshift__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rshift__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__and__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__or__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__xor__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__radd__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rsub__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rmul__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rmatmul__",
"attributes": [
{ "name": "self" },
{ "name": "other" }
]
},
{
"name": "__rtruediv__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rfloordiv__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rmod__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rpow__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rlshift__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rrshift__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rand__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ror__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__rxor__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__iadd__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__isub__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__imul__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__imatmul__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__itruediv__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ifloordiv__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__imod__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ipow__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ilshift__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__irshift__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__iand__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ior__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "__ixor__",
"attributes": [
{ "name": "self" },
{ "name": "value" }
]
},
{
"name": "transpose",
"attributes": [
{ "name": "self" }
],
"category": "Transform",
"varargs": "args"
},
{
"name": "astype",
"attributes": [
{ "name": "self" },
{ "name": "dtype" }
],
"category": "Layer"
},
{
"name": "reshape",
"attributes": [
{ "name": "self" }
],
"category": "Shape",
"varargs": "args"
},
{
"name": "_broadcast",
"attributes": [
{ "name": "self" }
],
"category": "Shape",
"varargs": "args"
},
{
"name": "flatten",
"attributes": [
{ "name": "self" }
],
"category": "Shape"
},
{
"name": "sum",
"attributes": [
{ "name": "self" },
{ "name": "axis" },
{ "name": "keepdims" }
]
},
{
"name": "prod",
"attributes": [
{ "name": "self" },
{ "name": "axis" },
{ "name": "keepdims" }
]
},
{
"name": "min",
"attributes": [
{ "name": "self" },
{ "name": "axis" },
{ "name": "keepdims" }
]
},
{
"name": "max",
"attributes": [
{ "name": "self" },
{ "name": "axis" },
{ "name": "keepdims" }
]
},
{
"name": "mean",
"attributes": [
{ "name": "self" },
{ "name": "axis" },
{ "name": "keepdims" }
]
},
{
"name": "__getitem__",
"attributes": [
{ "name": "self" },
{ "name": "index" }
]
},
{
"name": "__setitem__",
"attributes": [
{ "name": "self" },
{ "name": "index" },
{ "name": "value" }
]
},
{
"name": "megengine.module.adaptive_pooling.AdaptiveAvgPool2d",
"category": "Pool"
},
{
"name": "megengine.module.adaptive_pooling.AdaptiveMaxPool2d",
"category": "Pool"
},
{
"name": "megengine.module.pooling.AvgPool2d",
"category": "Pool"
},
{
"name": "megengine.module.batch_matmul_activation.BatchMatMulActivation",
"category": "Layer"
},
{
"name": "megengine.module.batchnorm.BatchNorm1d",
"category": "Normalization"
},
{
"name": "megengine.module.batchnorm.BatchNorm2d",
"category": "Normalization"
},
{
"name": "megengine.module.concat.Concat",
"category": "Tensor"
},
{
"name": "megengine.module.conv.Conv1d",
"category": "Layer"
},
{
"name": "megengine.module.conv.Conv2d",
"category": "Layer"
},
{
"name": "megengine.module.conv.Conv3d",
"category": "Layer"
},
{
"name": "megengine.module.conv_bn.ConvBn2d",
"category": "Layer"
},
{
"name": "megengine.module.conv_bn.ConvBnRelu2d",
"category": "Layer"
},
{
"name": "megengine.module.conv.ConvRelu2d",
"category": "Layer"
},
{
"name": "megengine.module.conv.ConvTranspose2d",
"category": "Layer"
},
{
"name": "megengine.module.conv.ConvTranspose3d",
"category": "Layer"
},
{
"name": "megengine.module.conv.DeformableConv2d",
"category": "Layer"
},
{
"name": "megengine.module.deformable_psroi_pooling.DeformablePSROIPooling",
"category": "Pool"
},
{
"name": "megengine.module.quant_dequant.DequantStub",
"category": "Layer"
},
{
"name": "megengine.module.dropout.Dropout",
"category": "Dropout"
},
{
"name": "megengine.module.elemwise.Elemwise"
},
{
"name": "megengine.module.embedding.Embedding",
"category": "Transform"
},
{
"name": "megengine.module.activation.GELU",
"category": "Activation"
},
{
"name": "megengine.module.normalization.GroupNorm",
"category": "Normalization"
},
{
"name": "megengine.module.identity.Identity",
"category": "Layer"
},
{
"name": "megengine.module.normalization.InstanceNorm",
"category": "Normalization"
},
{
"name": "megengine.module.rnn.LSTM",
"category": "Layer"
},
{
"name": "megengine.module.rnn.LSTMCell",
"category": "Layer"
},
{
"name": "megengine.module.normalization.LayerNorm",
"category": "Normalization"
},
{
"name": "megengine.module.activation.LeakyReLU",
"category": "Activation"
},
{
"name": "megengine.module.linear.Linear",
"category": "Layer"
},
{
"name": "megengine.module.conv.LocalConv2d",
"category": "Layer"
},
{
"name": "megengine.module.lrn.LocalResponseNorm",
"category": "Layer"
},
{
"name": "megengine.module.pooling.MaxPool2d",
"category": "Pool"
},
{
"name": "megengine.module.module.Module",
"category": "Layer"
},
{
"name": "megengine.module.activation.PReLU",
"category": "Activation"
},
{
"name": "megengine.module.padding.Pad",
"category": "Layer"
},
{
"name": "megengine.module.pixel_shuffle.PixelShuffle",
"category": "Layer"
},
{
"name": "megengine.module.quant_dequant.QuantStub",
"category": "Layer"
},
{
"name": "megengine.module.rnn.RNN",
"category": "Layer"
},
{
"name": "megengine.module.rnn.RNNCell",
"category": "Layer"
},
{
"name": "megengine.module.activation.ReLU",
"category": "Activation"
},
{
"name": "megengine.module.activation.SiLU",
"category": "Activation"
},
{
"name": "megengine.module.activation.Sigmoid",
"category": "Activation"
},
{
"name": "megengine.module.sliding_window.SlidingWindow",
"category": "Layer"
},
{
"name": "megengine.module.sliding_window.SlidingWindowTranspose",
"category": "Layer"
},
{
"name": "megengine.module.activation.Softmax",
"category": "Activation"
},
{
"name": "megengine.module.batchnorm.SyncBatchNorm",
"category": "Normalization"
},
{
"name": "megengine.module.qat.batch_matmul_activation.BatchMatMulActivation",
"category": "Activation"
},
{
"name": "megengine.module.qat.concat.Concat",
"category": "Tensor"
},
{
"name": "megengine.module.qat.conv.Conv2d",
"category": "Layer"
},
{
"name": "megengine.module.qat.conv_bn.ConvBn2d",
"category": "Layer"
},
{
"name": "megengine.module.qat.conv_bn.ConvBnRelu2d",
"category": "Layer"
},
{
"name": "megengine.module.qat.conv.ConvRelu2d",
"category": "Layer"
},
{
"name": "megengine.module.qat.conv.ConvTranspose2d",
"category": "Layer"
},
{
"name": "megengine.module.qat.quant_dequant.DequantStub",
"category": "Layer"
},
{
"name": "megengine.module.qat.linear.Linear",
"category": "Layer"
},
{
"name": "megengine.module.qat.module.QATModule",
"category": "Layer"
},
{
"name": "megengine.module.qat.quant_dequant.QuantStub",
"category": "Layer"
},
{
"name": "megengine.module.quantized.batch_matmul_activation.BatchMatMulActivation",
"category": "Activation"
},
{
"name": "megengine.module.quantized.concat.Concat",
"category": "Tensor"
},
{
"name": "megengine.module.quantized.conv.Conv2d",
"category": "Layer"
},
{
"name": "megengine.module.quantized.conv_bn.ConvBn2d",
"category": "Layer"
},
{
"name": "megengine.module.quantized.conv_bn.ConvBnRelu2d",
"category": "Layer"
},
{
"name": "megengine.module.quantized.conv.ConvRelu2d",
"category": "Layer"
},
{
"name": "megengine.module.quantized.conv.ConvTranspose2d",
"category": "Layer"
},
{
"name": "megengine.module.quantized.quant_dequant.DequantStub",
"category": "Layer"
},
{
"name": "megengine.module.quantized.linear.Linear",
"category": "Layer"
},
{
"name": "megengine.module.quantized.quant_dequant.QuantStub",
"category": "Layer"
},
{
"name": "megengine.module.quantized.module.QuantizedModule",
"category": "Layer"
},
{
"name": "Axis",
"attributes": [
{ "name": "axis", "type": "int32", "default": 0 }
]
},
{
"name": "ConvolutionV0",
"category": "Layer",
"attributes": [
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "data_type", "type": "ConvolutionV0DataType", "default": "FLOAT" },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" }
]
},
{
"name": "ConvolutionV1",
"category": "Layer",
"attributes": [
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" }
]
},
{
"name": "Convolution",
"category": "Layer",
"attributes": [
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" }
]
},
{
"name": "MaskPropagate",
"attributes": [
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "kernel_h", "type": "uint32", "default": 1 },
{ "name": "kernel_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 }
]
},
{
"name": "ConvPooling",
"category": "Pool",
"attributes": [
{ "name": "method", "type": "ConvPoolingMethod", "default": "WITH_TEXTURE_OBJ" },
{ "name": "convMode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "poolMode", "type": "ConvPoolingPoolMode", "default": "AVERAGE" },
{ "name": "nonlineMode", "type": "ConvPoolingNonlineMode", "default": "IDENTITY" },
{ "name": "pool_shape_h", "type": "uint32", "default": 1 },
{ "name": "pool_shape_w", "type": "uint32", "default": 1 },
{ "name": "pool_stride_h", "type": "uint32", "default": 1 },
{ "name": "pool_stride_w", "type": "uint32", "default": 1 },
{ "name": "pool_pad_h", "type": "uint32", "default": 0 },
{ "name": "pool_pad_w", "type": "uint32", "default": 0 },
{ "name": "conv_stride_h", "type": "uint32", "default": 1 },
{ "name": "conv_stride_w", "type": "uint32", "default": 1 },
{ "name": "conv_pad_h", "type": "uint32", "default": 0 },
{ "name": "conv_pad_w", "type": "uint32", "default": 0 }
]
},
{
"name": "ConvBiasV0",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "ConvBiasV1",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "data_type", "type": "ConvolutionV0DataType", "default": "FLOAT" },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "ConvBiasV2",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "ConvBiasV3",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "output_block_size", "type": "uint32", "default": 0 },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "ConvBias",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "SeparableConv",
"category": "Layer",
"attributes": [
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "borderMode", "type": "SeparableConvBorderMode", "default": "BORDER_REPLICATE" },
{ "name": "is_symm_kernel", "type": "bool", "default": true },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "ksize_h", "type": "uint32", "default": 3 },
{ "name": "ksize_w", "type": "uint32", "default": 3 },
{ "name": "anchor_h", "type": "uint32", "default": 1 },
{ "name": "anchor_w", "type": "uint32", "default": 1 }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "Images2Neibs",
"category": "Shape",
"attributes": [
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "window_h", "type": "uint32", "default": 3 },
{ "name": "window_w", "type": "uint32", "default": 3 }
]
},
{
"name": "SlidingWindowTranspose",
"category": "Transform",
"attributes": [
{ "name": "out_h", "type": "uint32", "default": 0 },
{ "name": "out_w", "type": "uint32", "default": 0 },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "window_h", "type": "uint32", "default": 3 },
{ "name": "window_w", "type": "uint32", "default": 3 }
]
},
{
"name": "PoolingV0",
"category": "Pool",
"attributes": [
{ "name": "mode", "type": "PoolingV0Mode", "default": "MAX_" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 2 },
{ "name": "stride_w", "type": "uint32", "default": 2 },
{ "name": "window_h", "type": "uint32", "default": 2 },
{ "name": "window_w", "type": "uint32", "default": 2 },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" }
]
},
{
"name": "PoolingV1",
"category": "Pool"
},
{
"name": "Pooling",
"category": "Pool",
"attributes": [
{ "name": "mode", "type": "PoolingV0Mode", "default": "MAX_" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 2 },
{ "name": "stride_w", "type": "uint32", "default": 2 },
{ "name": "window_h", "type": "uint32", "default": 2 },
{ "name": "window_w", "type": "uint32", "default": 2 },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" }
]
},
{
"name": "Softmax",
"category": "Activation",
"attributes": [
{ "name": "axis", "type": "int32", "default": -1 }
]
},
{
"name": "AdaptivePoolingV0",
"category": "Activation",
"attributes": [
{ "name": "mode", "type": "PoolingV0Mode", "default": "MAX_" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" }
]
},
{
"name": "AdaptivePooling",
"category": "Activation",
"attributes": [
{ "name": "mode", "type": "PoolingV0Mode", "default": "MAX_" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" }
]
},
{
"name": "LRN",
"category": "Normalization",
"attributes": [
{ "name": "n", "type": "uint32", "default": 5 },
{ "name": "k", "type": "float32", "default": 2 },
{ "name": "alpha", "type": "float32", "default": 0.0001 },
{ "name": "beta", "type": "float32", "default": 0.75 }
]
},
{
"name": "BN",
"category": "Normalization",
"attributes": [
{ "name": "param_dim", "type": "BNParamDim", "default": "DIM_11HW" },
{ "name": "fwd_mode", "type": "BNFwdMode", "default": "TRAINING" },
{ "name": "epsilon", "type": "float64", "default": 0.0001 },
{ "name": "avg_factor", "type": "float64", "default": 1 },
{ "name": "scale", "type": "float32", "default": 1 },
{ "name": "bias", "type": "float32", "default": 0 }
]
},
{
"name": "ROIPooling",
"attributes": [
{ "name": "mode", "type": "ROIPoolingMode", "default": "MAX_" },
{ "name": "scale", "type": "float32", "default": 1 }
]
},
{
"name": "WarpPerspectiveV1",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "bmode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "border_val", "type": "float32", "default": 0 }
]
},
{
"name": "WarpPerspective",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "bmode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "border_val", "type": "float32", "default": 0 }
]
},
{
"name": "SpatialTfGridGenerator",
"attributes": [
{ "name": "mode", "type": "SpatialTfGridGeneratorMode", "default": "AFFINE" }
]
},
{
"name": "SpatialTfSampler",
"attributes": [
{ "name": "mode", "type": "SpatialTfSamplerMode", "default": "BILINEAR" }
]
},
{
"name": "AddUpdate",
"attributes": [
{ "name": "alpha", "type": "float32", "default": 1 },
{ "name": "beta", "type": "float32", "default": 1 },
{ "name": "bias", "type": "float32", "default": 0 }
]
},
{
"name": "Elemwise",
"attributes": [
{ "name": "mode", "type": "ElemwiseMode", "default": "RELU" }
],
"inputs": [
{ "name": "A" },
{ "name": "B" }
]
},
{
"name": "ElemwiseMultiType",
"attributes": [
{ "name": "mode", "type": "ElemwiseMultiTypeMode", "default": "FUSE_MUL_ADD3_INT16x32x32x32" }
]
},
{
"name": "PowC",
"attributes": [
{ "name": "exp", "type": "float32", "default": 0 }
]
},
{
"name": "DctChannelSelectV0",
"attributes": [
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "fastImpl", "type": "DctChannelSelectV0FastImpl", "default": "NONE" },
{ "name": "dct_block_size", "type": "int32", "default": 8 }
]
},
{
"name": "DctChannelSelect",
"attributes": [
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "fastImpl", "type": "DctChannelSelectV0FastImpl", "default": "NONE" },
{ "name": "dct_block_size", "type": "int32", "default": 8 }
]
},
{
"name": "MatrixMulV0",
"attributes": [
{ "name": "transposeA", "type": "bool", "default": false },
{ "name": "transposeB", "type": "bool", "default": false },
{ "name": "data_type", "type": "MatrixMulV0DataType", "default": "FLOAT" }
],
"inputs": [
{ "name": "A" },
{ "name": "B" }
]
},
{
"name": "MatrixMulV1",
"attributes": [
{ "name": "transposeA", "type": "bool", "default": false },
{ "name": "transposeB", "type": "bool", "default": false },
{ "name": "compute_mode", "type": "MatrixMulV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "A" },
{ "name": "B" }
]
},
{
"name": "MatrixMulV2",
"attributes": [
{ "name": "transposeA", "type": "bool", "default": false },
{ "name": "transposeB", "type": "bool", "default": false },
{ "name": "compute_mode", "type": "MatrixMulV1ComputeMode", "default": "DEFAULT" },
{ "name": "format", "type": "MatrixMulFormat", "default": "DEFAULT" }
],
"inputs": [
{ "name": "A" },
{ "name": "B" }
]
},
{
"name": "MatrixMul",
"attributes": [
{ "name": "transposeA", "type": "bool", "default": false },
{ "name": "transposeB", "type": "bool", "default": false },
{ "name": "compute_mode", "type": "MatrixMulV1ComputeMode", "default": "DEFAULT" },
{ "name": "format", "type": "MatrixMulFormat", "default": "DEFAULT" }
],
"inputs": [
{ "name": "A" },
{ "name": "B" }
]
},
{
"name": "SVD",
"attributes": [
{ "name": "full_matrices", "type": "bool", "default": false },
{ "name": "compute_uv", "type": "bool", "default": true }
]
},
{
"name": "ReduceV0",
"attributes": [
{ "name": "mode", "type": "ReduceV0Mode", "default": "SUM" },
{ "name": "axis", "type": "int32", "default": -1 }
]
},
{
"name": "ReduceV1",
"attributes": [
{ "name": "mode", "type": "ReduceV1Mode", "default": "SUM" },
{ "name": "axis", "type": "int32", "default": -1 },
{ "name": "data_type", "type": "ReduceV1DataType", "default": "DEFAULT" }
]
},
{
"name": "Reduce",
"attributes": [
{ "name": "mode", "type": "ReduceMode", "default": "SUM" },
{ "name": "axis", "type": "int32", "default": 2147483647 },
{ "name": "data_type", "type": "ReduceDataType", "default": "DEFAULT" }
]
},
{
"name": "CumsumV0",
"attributes": [
{ "name": "axis", "type": "int32", "default": -1 },
{ "name": "exclusive", "type": "bool", "default": true },
{ "name": "reverse", "type": "bool", "default": false }
]
},
{
"name": "Cumsum",
"attributes": [
{ "name": "axis", "type": "int32", "default": 2147483647 },
{ "name": "exclusive", "type": "bool", "default": true },
{ "name": "reverse", "type": "bool", "default": false }
]
},
{
"name": "CondTake",
"attributes": [
{ "name": "mode", "type": "CondTakeMode", "default": "EQ" },
{ "name": "val", "type": "float32", "default": 0 },
{ "name": "eps", "type": "float32", "default": 0.000001 }
]
},
{
"name": "Argsort",
"attributes": [
{ "name": "order", "type": "ArgsortOrder", "default": "ASCENDING" }
]
},
{
"name": "IndexingRemap",
"attributes": [
{ "name": "is_non_overlapping", "type": "bool", "default": false }
]
},
{
"name": "Sleep",
"attributes": [
{ "name": "time", "type": "float32", "default": 0 }
]
},
{
"name": "Linspace",
"attributes": [
{ "name": "endpoint", "type": "bool", "default": true }
]
},
{
"name": "LinspaceFull",
"attributes": [
{ "name": "start", "type": "float64", "default": 0 },
{ "name": "stop", "type": "float64", "default": 1 },
{ "name": "endpoint", "type": "bool", "default": true }
]
},
{
"name": "Eye",
"attributes": [
{ "name": "k", "type": "int32", "default": 0 },
{ "name": "dtype", "type": "DTypeEnum", "default": "Float32" }
]
},
{
"name": "Diag",
"attributes": [
{ "name": "k", "type": "int32", "default": 0 }
]
},
{
"name": "UniformRNGV0",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 }
]
},
{
"name": "UniformRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 },
{ "name": "dtype", "type": "DTypeEnum", "default": "Float32" }
]
},
{
"name": "GaussianRNGV0",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 },
{ "name": "mean", "type": "float32", "default": 0 },
{ "name": "std", "type": "float32", "default": 1 }
]
},
{
"name": "GaussianRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 },
{ "name": "mean", "type": "float32", "default": 0 },
{ "name": "std", "type": "float32", "default": 1 },
{ "name": "dtype", "type": "DTypeEnum", "default": "Float32" }
]
},
{
"name": "GammaRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 }
]
},
{
"name": "BetaRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 }
]
},
{
"name": "PoissonRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 }
]
},
{
"name": "PermutationRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 },
{ "name": "dtype", "type": "DTypeEnum", "default": "Int32" }
]
},
{
"name": "ShuffleRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 }
]
},
{
"name": "Flip",
"category": "Shape",
"attributes": [
{ "name": "vertical", "type": "bool", "default": false },
{ "name": "horizontal", "type": "bool", "default": false }
]
},
{
"name": "Rotate",
"attributes": [
{ "name": "clockwise", "type": "bool", "default": true }
]
},
{
"name": "ROICopy",
"attributes": [
{ "name": "row_from", "type": "uint32", "default": 0 },
{ "name": "row_to", "type": "uint32", "default": 0 },
{ "name": "col_from", "type": "uint32", "default": 0 },
{ "name": "col_to", "type": "uint32", "default": 0 }
]
},
{
"name": "CvtColor",
"attributes": [
{ "name": "mode", "type": "CvtColorMode", "default": "RGB2GRAY" }
]
},
{
"name": "WarpAffineV0",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "border_mode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "border_val", "type": "float32", "default": 0 }
]
},
{
"name": "WarpAffineV1",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "border_mode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "border_val", "type": "float32", "default": 0 },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NHWC" }
]
},
{
"name": "WarpAffine",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "border_mode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "border_val", "type": "float32", "default": 0 },
{ "name": "format", "type": "ConvolutionFormat", "default": "NHWC" }
]
},
{
"name": "GaussianBlur",
"attributes": [
{ "name": "border_mode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "kernel_height", "type": "uint32", "default": 0 },
{ "name": "kernel_width", "type": "uint32", "default": 0 },
{ "name": "sigma_x", "type": "float32", "default": 0 },
{ "name": "sigma_y", "type": "float32", "default": 0 }
]
},
{
"name": "ResizeV0",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" }
]
},
{
"name": "ResizeV1",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NHWC" }
]
},
{
"name": "Resize",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NHWC" }
]
},
{
"name": "RemapV0",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "border_type", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NHWC" },
{ "name": "scalar", "type": "float32", "default": 0 }
]
},
{
"name": "Remap",
"attributes": [
{ "name": "imode", "type": "WarpPerspectiveV1InterpolationMode", "default": "LINEAR" },
{ "name": "border_type", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NHWC" },
{ "name": "scalar", "type": "float32", "default": 0 }
]
},
{
"name": "Convolution3D",
"category": "Layer",
"attributes": [
{ "name": "mode", "type": "Convolution3DMode", "default": "CROSS_CORRELATION" },
{ "name": "pad_d", "type": "uint32", "default": 0 },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_d", "type": "uint32", "default": 1 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_d", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "Convolution3DSparse", "default": "DENSE" },
{ "name": "data_type", "type": "Convolution3DDataType", "default": "FLOAT" },
{ "name": "format", "type": "Convolution3DFormat", "default": "NCDHW" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" }
]
},
{
"name": "Conv3DBias",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "Conv3DBiasNonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "Convolution3DMode", "default": "CROSS_CORRELATION" },
{ "name": "pad_d", "type": "uint32", "default": 0 },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_d", "type": "uint32", "default": 1 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 0 }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "SeparableConv3D",
"category": "Layer",
"attributes": [
{ "name": "mode", "type": "Convolution3DMode", "default": "CROSS_CORRELATION" },
{ "name": "borderMode", "type": "SeparableConv3DBorderMode", "default": "BORDER_REPLICATE" },
{ "name": "is_symm_kernel", "type": "bool", "default": true },
{ "name": "pad_d", "type": "uint32", "default": 0 },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_d", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "ksize_d", "type": "uint32", "default": 0 },
{ "name": "ksize_h", "type": "uint32", "default": 3 },
{ "name": "ksize_w", "type": "uint32", "default": 3 },
{ "name": "anchor_d", "type": "uint32", "default": 0 },
{ "name": "anchor_h", "type": "uint32", "default": 1 },
{ "name": "anchor_w", "type": "uint32", "default": 1 }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "TopK",
"attributes": [
{ "name": "mode", "type": "TopKMode", "default": "KTH_ONLY" }
]
},
{
"name": "RelayoutFormatV0",
"attributes": [
{ "name": "mode", "type": "RelayoutFormatV0Mode", "default": "NHWC_NHWCD4" }
]
},
{
"name": "RelayoutFormat",
"attributes": [
{ "name": "mode", "type": "RelayoutFormatV0Mode", "default": "NHWC_NHWCD4" },
{ "name": "oc", "type": "uint32", "default": 0 },
{ "name": "group", "type": "uint32", "default": 1 }
]
},
{
"name": "SeparableFilterV0",
"attributes": [
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "borderMode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "is_symm_kernel", "type": "bool", "default": true },
{ "name": "ksize_h", "type": "uint32", "default": 3 },
{ "name": "ksize_w", "type": "uint32", "default": 3 },
{ "name": "anchor_h", "type": "uint32", "default": 1 },
{ "name": "anchor_w", "type": "uint32", "default": 1 }
]
},
{
"name": "SeparableFilter",
"attributes": [
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "borderMode", "type": "WarpPerspectiveV1BorderMode", "default": "REPLICATE" },
{ "name": "is_symm_kernel", "type": "bool", "default": true },
{ "name": "ksize_h", "type": "uint32", "default": 3 },
{ "name": "ksize_w", "type": "uint32", "default": 3 },
{ "name": "anchor_h", "type": "uint32", "default": 1 },
{ "name": "anchor_w", "type": "uint32", "default": 1 }
]
},
{
"name": "LocalShareV0",
"attributes": [
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "spatial_groups_h", "type": "uint32", "default": 1 },
{ "name": "spatial_groups_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "computeMode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
]
},
{
"name": "LocalShare",
"attributes": [
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "spatial_groups_h", "type": "uint32", "default": 1 },
{ "name": "spatial_groups_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "computeMode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
]
},
{
"name": "ROIAlignV0",
"attributes": [
{ "name": "mode", "type": "ROIAlignV0Mode", "default": "MAX_" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "spatial_scale", "type": "float32", "default": 1 },
{ "name": "offset", "type": "float32", "default": 0 },
{ "name": "pooled_height", "type": "uint32", "default": 1 },
{ "name": "pooled_width", "type": "uint32", "default": 1 },
{ "name": "sample_height", "type": "uint32", "default": 2 },
{ "name": "sample_width", "type": "uint32", "default": 2 }
]
},
{
"name": "ROIAlign",
"attributes": [
{ "name": "mode", "type": "ROIAlignV0Mode", "default": "MAX_" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "spatial_scale", "type": "float32", "default": 1 },
{ "name": "offset", "type": "float32", "default": 0 },
{ "name": "pooled_height", "type": "uint32", "default": 1 },
{ "name": "pooled_width", "type": "uint32", "default": 1 },
{ "name": "sample_height", "type": "uint32", "default": 2 },
{ "name": "sample_width", "type": "uint32", "default": 2 }
]
},
{
"name": "Correlation",
"attributes": [
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "kernel_size", "type": "uint32", "default": 1 },
{ "name": "max_displacement", "type": "uint32", "default": 1 },
{ "name": "stride1", "type": "uint32", "default": 1 },
{ "name": "stride2", "type": "uint32", "default": 1 },
{ "name": "pad_size", "type": "uint32", "default": 0 },
{ "name": "is_multiply", "type": "bool", "default": true }
]
},
{
"name": "DeformablePSROIPooling",
"attributes": [
{ "name": "no_trans", "type": "bool", "default": true },
{ "name": "spatial_scale", "type": "float32", "default": 1 },
{ "name": "trans_std", "type": "float32", "default": 1 },
{ "name": "pooled_h", "type": "uint32", "default": 1 },
{ "name": "pooled_w", "type": "uint32", "default": 1 },
{ "name": "part_size", "type": "uint32", "default": 1 },
{ "name": "sample_per_part", "type": "uint32", "default": 1 }
]
},
{
"name": "BatchConvBiasV0",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionV0Format", "default": "NCHW" },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "BatchConvBias",
"attributes": [
{ "name": "nonlineMode", "type": "ConvBiasV0NonlineMode", "default": "IDENTITY" },
{ "name": "mode", "type": "ConvolutionV0Mode", "default": "CROSS_CORRELATION" },
{ "name": "pad_h", "type": "uint32", "default": 0 },
{ "name": "pad_w", "type": "uint32", "default": 0 },
{ "name": "stride_h", "type": "uint32", "default": 1 },
{ "name": "stride_w", "type": "uint32", "default": 1 },
{ "name": "dilate_h", "type": "uint32", "default": 1 },
{ "name": "dilate_w", "type": "uint32", "default": 1 },
{ "name": "sparse", "type": "ConvolutionV0Sparse", "default": "DENSE" },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" },
{ "name": "compute_mode", "type": "ConvolutionV1ComputeMode", "default": "DEFAULT" }
],
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "FakeQuant",
"attributes": [
{ "name": "qmin", "type": "int32", "default": -2147483648 },
{ "name": "qmax", "type": "int32", "default": 2147483647 }
]
},
{
"name": "TQT",
"category": "Quantization",
"attributes": [
{ "name": "qmin", "type": "int32", "default": -2147483648 },
{ "name": "qmax", "type": "int32", "default": 2147483647 }
]
},
{
"name": "LSQ",
"category": "Quantization",
"attributes": [
{ "name": "qmin", "type": "int32", "default": -2147483648 },
{ "name": "qmax", "type": "int32", "default": 2147483647 }
]
},
{
"name": "Fill",
"attributes": [
{ "name": "value", "type": "float32", "default": 0 }
]
},
{
"name": "CheckNonFinite",
"attributes": [
{ "name": "scale", "type": "float32", "default": 1 }
]
},
{
"name": "Padding",
"category": "Layer",
"attributes": [
{ "name": "front_offset_dim0", "type": "uint32", "default": 0 },
{ "name": "front_offset_dim1", "type": "uint32", "default": 0 },
{ "name": "front_offset_dim2", "type": "uint32", "default": 0 },
{ "name": "front_offset_dim3", "type": "uint32", "default": 0 },
{ "name": "front_offset_dim4", "type": "uint32", "default": 0 },
{ "name": "front_offset_dim5", "type": "uint32", "default": 0 },
{ "name": "front_offset_dim6", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim0", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim1", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim2", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim3", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim4", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim5", "type": "uint32", "default": 0 },
{ "name": "back_offset_dim6", "type": "uint32", "default": 0 },
{ "name": "padding_val", "type": "float32", "default": 0 },
{ "name": "padding_mode", "type": "PaddingPaddingMode", "default": "CONSTANT" }
]
},
{
"name": "LayerNorm",
"category": "Normalization",
"attributes": [
{ "name": "affine", "type": "bool", "default": true },
{ "name": "eps", "type": "float32", "default": 0.00001 },
{ "name": "normalized_dim", "type": "uint64", "default": 1 },
{ "name": "normalized_size", "type": "uint64", "default": 1 }
]
},
{
"name": "Dropout",
"category": "Dropout",
"attributes": [
{ "name": "drop_prob", "type": "float32", "default": 0 },
{ "name": "seed", "type": "uint64", "default": 0 }
]
},
{
"name": "RNNCell",
"category": "Layer",
"attributes": [
{ "name": "nonlineMode", "type": "RNNCellNonlineMode", "default": "IDENTITY" }
]
},
{
"name": "RNN",
"category": "Layer",
"attributes": [
{ "name": "num_layers", "type": "uint32", "default": 1 },
{ "name": "bidirectional", "type": "bool", "default": false },
{ "name": "bias", "type": "bool", "default": true },
{ "name": "hidden_size", "type": "uint32", "default": 128 },
{ "name": "dropout", "type": "float32", "default": 0 },
{ "name": "nonlineMode", "type": "RNNCellNonlineMode", "default": "IDENTITY" },
{ "name": "fwd_mode", "type": "BNFwdMode", "default": "TRAINING" }
]
},
{
"name": "LSTM",
"category": "Layer",
"attributes": [
{ "name": "num_layers", "type": "uint32", "default": 1 },
{ "name": "bidirectional", "type": "bool", "default": false },
{ "name": "bias", "type": "bool", "default": true },
{ "name": "hidden_size", "type": "uint32", "default": 128 },
{ "name": "proj_size", "type": "uint32", "default": 0 },
{ "name": "dropout", "type": "float32", "default": 0 },
{ "name": "fwd_mode", "type": "BNFwdMode", "default": "TRAINING" }
]
},
{
"name": "DType",
"attributes": [
{ "name": "dtype", "type": "DTypeEnum", "default": "Byte" }
]
},
{
"name": "PersistentOutputStorage",
"attributes": [
{ "name": "share_key", "type": "int32", "default": -1 }
]
},
{
"name": "OptionalAxis",
"attributes": [
{ "name": "axis", "type": "int32", "default": -1 }
]
},
{
"name": "OptionalAxisV1",
"attributes": [
{ "name": "axis", "type": "int32", "default": 7 }
]
},
{
"name": "ExecutionPolicyV0",
"attributes": [
{ "name": "strategy", "type": "ExecutionPolicyV0Strategy", "default": "HEURISTIC" },
{ "name": "workspace_limit", "type": "uint64", "default": 18446744073709552000 }
]
},
{
"name": "ExecutionPolicy",
"attributes": [
{ "name": "strategy", "type": "ExecutionPolicyStrategy", "default": "PROFILE" },
{ "name": "workspace_limit", "type": "uint64", "default": 18446744073709552000 }
]
},
{
"name": "AssertEqual",
"attributes": [
{ "name": "maxerr", "type": "float32", "default": 0.0001 },
{ "name": "verbose", "type": "bool", "default": false }
]
},
{
"name": "FpgaConv",
"attributes": [
{ "name": "need_output_quantize", "type": "bool", "default": false },
{ "name": "need_output_threshold", "type": "bool", "default": false },
{ "name": "stride", "type": "int32", "default": 1 },
{ "name": "input_bit_width", "type": "int32", "default": 2 },
{ "name": "output_bit_width", "type": "int32", "default": 2 },
{ "name": "weight_bit_width", "type": "int32", "default": 2 },
{ "name": "thres0", "type": "int32", "default": 0 },
{ "name": "thres1", "type": "int32", "default": 1 },
{ "name": "unpool_size", "type": "uint32", "default": 4 },
{ "name": "direct_size", "type": "uint32", "default": 4 }
]
},
{
"name": "CollectiveComm",
"attributes": [
{ "name": "mode", "type": "CollectiveCommMode", "default": "REDUCE_SUM" }
]
},
{
"name": "CondExecPred",
"attributes": [
{ "name": "mode", "type": "CondExecPredMode", "default": "CASE" },
{ "name": "eps", "type": "float32", "default": 0.0001 }
]
},
{
"name": "CondExecPredLogical",
"attributes": [
{ "name": "mode", "type": "CondExecPredLogicalMode", "default": "OR" }
]
},
{
"name": "CondExecMark",
"attributes": [
{ "name": "grad_mode", "type": "CondExecMarkGradMode", "default": "SUM" },
{ "name": "static_infer", "type": "CondExecMarkStaticInfer", "default": "SHAPE_VALUE" }
]
},
{
"name": "CondExecMerge",
"attributes": [
{ "name": "nr_output", "type": "uint32", "default": 1 },
{ "name": "mode", "type": "CondExecMergeMode", "default": "EXACT_ONE" }
]
},
{
"name": "NvOf",
"attributes": [
{ "name": "precision", "type": "uint32", "default": 1 }
]
},
{
"name": "PersistentDTypeScalar",
"attributes": [
{ "name": "dtype", "type": "DTypeEnum", "default": "Float32" },
{ "name": "storage", "type": "uint8[]", "default": 0 }
]
},
{
"name": "MGBAddUpdate",
"attributes": [
{ "name": "alpha", "type": "PersistentDTypeScalar" },
{ "name": "beta", "type": "PersistentDTypeScalar" },
{ "name": "bias", "type": "PersistentDTypeScalar" }
]
},
{
"name": "Host2DeviceCopy",
"category": "Data",
"attributes": [
{ "name": "enable_value_infer", "type": "bool", "default": true },
{ "name": "dump_default_value", "type": "bool", "default": false },
{ "name": "allow_cpu_mem_fwd", "type": "bool", "default": true }
]
},
{
"name": "Dimshuffle",
"category": "Shape",
"attributes": [
{ "name": "pattern", "type": "int32[]", "default": 0 },
{ "name": "ndim", "type": "uint32", "default": 0 }
]
},
{
"name": "AxisDesc",
"attributes": [
{ "name": "method", "type": "AxisDescMethod", "default": "ADD_1" },
{ "name": "axis", "type": "int32", "default": 0 }
]
},
{
"name": "AxisAddRemove",
"attributes": [
{ "name": "desc", "type": "AxisDesc[]" }
]
},
{
"name": "MGBSleep",
"attributes": [
{ "name": "device", "type": "bool", "default": true },
{ "name": "host", "type": "bool", "default": false },
{ "name": "seconds", "type": "float64", "default": 0 }
]
},
{
"name": "IndexDescMaskItem",
"attributes": [
{ "name": "axis", "type": "int8", "default": 0 },
{ "name": "begin", "type": "bool", "default": false },
{ "name": "end", "type": "bool", "default": false },
{ "name": "step", "type": "bool", "default": false },
{ "name": "idx", "type": "bool", "default": false }
]
},
{
"name": "IndexDescMaskDump",
"attributes": [
{ "name": "items", "type": "IndexDescMaskItem[]" }
]
},
{
"name": "NMSKeep",
"attributes": [
{ "name": "iou_thresh", "type": "float32", "default": 0 },
{ "name": "max_output", "type": "uint32", "default": 0 }
]
},
{
"name": "BatchNormForward",
"category": "Normalization"
},
{
"name": "ConvolutionForward",
"category": "Layer",
"inputs": [
{ "name": "input" },
{ "name": "weights" }
]
},
{
"name": "ConvBiasForward",
"category": "Layer",
"inputs": [
{ "name": "input" },
{ "name": "weights" },
{ "name": "biases" }
]
},
{
"name": "PoolingForward",
"category": "Pool"
},
{
"name": "AdaptivePoolingForward",
"category": "Pool"
},
{
"name": "Subtensor",
"inputs": [
{ "name": "input" },
{ "name": "axis" }
]
},
{
"name": "GetVarShape",
"category": "Shape"
},
{
"name": "GetVarShapeV1",
"category": "Shape"
},
{
"name": "ReshapeV1",
"category": "Shape",
"inputs": [
{ "name": "tensor" },
{ "name": "target_shape" }
],
"outputs": [
{ "name": "tensor" }
]
},
{
"name": "Reshape",
"category": "Shape",
"inputs": [
{ "name": "tensor" },
{ "name": "target_shape" }
],
"outputs": [
{ "name": "tensor" }
]
},
{
"name": "Concat",
"category": "Tensor"
},
{
"name": "GroupNorm",
"attributes": [
{ "name": "affine", "type": "bool", "default": true },
{ "name": "eps", "type": "float32", "default": 0.00001 },
{ "name": "group", "type": "uint32", "default": 1 },
{ "name": "format", "type": "ConvolutionFormat", "default": "NCHW" }
]
},
{
"name": "MultinomialRNG",
"attributes": [
{ "name": "seed", "type": "uint64", "default": 0 },
{ "name": "num_samples", "type": "uint64", "default": 1 },
{ "name": "replacement", "type": "bool", "default": false }
]
}
]
Reference in New Issue View Git Blame Copy Permalink