7254f7b4d1
Build / Build (macos-latest) (push) Has been cancelled
Build / Build (ubuntu-latest) (push) Has been cancelled
Build / Build (windows-latest) (push) Has been cancelled
Build / Analyze (javascript) (push) Has been cancelled
Build / Analyze (python) (push) Has been cancelled
1051 lines
29 KiB
JSON
1051 lines
29 KiB
JSON
[
|
|
{
|
|
"name": "input_layer",
|
|
"description": "Represents an input of the model",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "output_layer",
|
|
"description": "Represents an output of the model",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "postprocess",
|
|
"description": "Represents a whole post-processing function of some meta-architecture",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "max_proposals_per_class",
|
|
"type": "int64",
|
|
"description": "Maximum number of proposals per class",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "iou_th",
|
|
"type": "float32",
|
|
"visible": false,
|
|
"description": "Intersection over union overlap threshold, used in the NMS iterative elimination process where potential duplicates of detected items are ignored"
|
|
},
|
|
{
|
|
"name": "meta_arch",
|
|
"type": "string",
|
|
"visible": false,
|
|
"description": "Postprocessing meta-architecture name"
|
|
},
|
|
{
|
|
"name": "max_total_output_proposals",
|
|
"type": "int64",
|
|
"visible": false,
|
|
"description": "Maximum number of bounding box proposals"
|
|
},
|
|
{
|
|
"name": "postprocess_type",
|
|
"type": "string",
|
|
"visible": false,
|
|
"description": "Postprocessing type name"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "conv",
|
|
"category": "Layer",
|
|
"description": "Convolution layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "strides",
|
|
"type": "int64[]",
|
|
"description": "Stride along each axis (batch, height, width, features)"
|
|
},
|
|
{
|
|
"name": "dilations",
|
|
"type": "int64[]",
|
|
"description": "Dilation value along each axis (batch, height, width, features)"
|
|
},
|
|
{
|
|
"name": "padding",
|
|
"type": "string",
|
|
"description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV"
|
|
},
|
|
{
|
|
"name": "groups",
|
|
"type": "int64",
|
|
"description": "Number of groups input channels and output channels are divided into"
|
|
},
|
|
{
|
|
"name": "batch_norm",
|
|
"type": "boolean",
|
|
"description": "Whether batch normalization is folded into the layer"
|
|
},
|
|
{
|
|
"name": "elementwise_add",
|
|
"type": "boolean",
|
|
"description": "Whether elementwise addition is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "pre_layer_batch_norm",
|
|
"type": "boolean",
|
|
"description": "Whether batch normalization is folded into the layer, before the operation itself",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "transpose_output_width_features",
|
|
"type": "boolean",
|
|
"description": "Whether to transpose the width and the features axes of the layer's output tensor",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "spatial_flatten_output",
|
|
"type": "boolean",
|
|
"description": "Whether to flatten the layer's output to one row",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "relu",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "delta",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"category": "Activation",
|
|
"description": "Activation function",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "batch_norm",
|
|
"type": "boolean",
|
|
"description": "Whether batch normalization is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "elementwise_add",
|
|
"type": "boolean",
|
|
"description": "Whether elementwise addition is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "argmax",
|
|
"description": "Argmax layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "avgpool",
|
|
"category": "Pool",
|
|
"description": "Average pooling layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "strides",
|
|
"type": "int64[]",
|
|
"description": "Stride along each axis (batch, height, width, features)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "padding",
|
|
"type": "string",
|
|
"description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "batch_norm",
|
|
"category": "Normalization",
|
|
"description": "Batch normalization layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "elementwise_add",
|
|
"type": "boolean",
|
|
"description": "Whether elementwise addition is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "bbox_decoder",
|
|
"description": "Bounding box decoding layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "deconv",
|
|
"category": "Layer",
|
|
"description": "Deconvolution (transposed convolution) layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "strides",
|
|
"type": "int64[]",
|
|
"description": "Stride along each axis (batch, height, width, features)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "dilations",
|
|
"type": "int64[]",
|
|
"description": "Dilation value along each axis (batch, height, width, features)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "padding",
|
|
"type": "string",
|
|
"description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "groups",
|
|
"type": "int64",
|
|
"description": "Number of groups input channels and output channels are divided into",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "batch_norm",
|
|
"type": "boolean",
|
|
"description": "Whether batch normalization is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "elementwise_add",
|
|
"type": "boolean",
|
|
"description": "Whether elementwise addition is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "dense",
|
|
"category": "Layer",
|
|
"description": "Dense (fully connected) layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "batch_norm",
|
|
"type": "boolean",
|
|
"description": "Whether batch normalization is folded into the layer",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "depth_to_space",
|
|
"description": "Depth to space layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "block_sizes",
|
|
"type": "int64[]",
|
|
"description": "Block size along each spatial axis",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "depth_to_space_type",
|
|
"type": "string",
|
|
"description": "Depth to space variant, either dcr (depth-column-row) or crd (column-row-depth)",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "dw",
|
|
"description": "Depthwise convolution layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "strides",
|
|
"type": "int64[]",
|
|
"description": "Stride along each axis (batch, height, width, features)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "dilations",
|
|
"type": "int64[]",
|
|
"description": "Dilation value along each axis (batch, height, width, features)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "padding",
|
|
"type": "string",
|
|
"description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "groups",
|
|
"type": "int64",
|
|
"description": "Number of groups input channels and output channels are divided into",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "batch_norm",
|
|
"type": "boolean",
|
|
"description": "Whether batch normalization is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "elementwise_add",
|
|
"type": "boolean",
|
|
"description": "Whether elementwise addition is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "transpose_output_width_features",
|
|
"type": "string",
|
|
"description": "Whether to transpose the width and the features axes of the layer's output tensor",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "dynamic_weights",
|
|
"type": "boolean",
|
|
"description": "Whether the layer's weights are data driven",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "external_pad",
|
|
"description": "Padding layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "external_pad_params",
|
|
"type": "int64[]",
|
|
"description": "Padding value in pixels in each edge (top, bottom, left, right)",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "feature_interleave",
|
|
"description": "Feature interleave layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "feature_multiplier",
|
|
"description": "Elementwise feature multiplication layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "feature_multiplier_type",
|
|
"type": "string",
|
|
"description": "Feature multiplier variant, either square (to multiply each value by itself), or user_specified",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "feature_shuffle",
|
|
"description": "Feature shuffle layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "format_conversion",
|
|
"description": "Reshapes the input tensor between different memory layouts",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "expand_spatial_sizes",
|
|
"type": "int64[]",
|
|
"description": "New output tensor dimensions after the reshape (height, width)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "conversion_type",
|
|
"type": "string",
|
|
"visible": false,
|
|
"description": "Format conversion variant"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "global_avg_pool",
|
|
"category": "Pool",
|
|
"description": "Global average pooling layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "maxpool",
|
|
"category": "Pool",
|
|
"description": "Maximum pooling layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "strides",
|
|
"type": "int64[]",
|
|
"description": "Stride along each axis (batch, height, width, features)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "padding",
|
|
"type": "string",
|
|
"description": "Padding mode, either VALID, SAME (symmetric, Caffe-like), SAME_TENSORFLOW, or DECONV",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "nms",
|
|
"description": "Non-maximum suppression layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "scores_threshold",
|
|
"type": "float32",
|
|
"description": "Confidence threshold for NMS filtering",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "iou_threshold",
|
|
"type": "float32",
|
|
"description": "Intersection over union overlap threshold, used in the NMS iterative elimination process where potential duplicates of detected items are ignored",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "classes",
|
|
"type": "int64",
|
|
"description": "Number of NMS classes",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "max_output_size",
|
|
"type": "int64",
|
|
"description": "Maximum number of proposals per class",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "normalization",
|
|
"category": "Normalization",
|
|
"description": "Normalization layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "elementwise_add",
|
|
"type": "boolean",
|
|
"description": "Whether elementwise addition is folded into the layer",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "proposal_generator",
|
|
"description": "Proposal generator layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "reduce_l2",
|
|
"description": "Reduce layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "reduce_max",
|
|
"description": "Reduce Max layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "groups",
|
|
"type": "int64",
|
|
"description": "Number of groups input channels and output channels are divided into",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "reduce_sum",
|
|
"description": "Reduce Sum layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "groups",
|
|
"type": "int64",
|
|
"description": "Number of groups input channels and output channels are divided into",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "reduce_axes",
|
|
"type": "int64[]",
|
|
"description": "List of axes to reduce",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "resize",
|
|
"category": "Tensor",
|
|
"description": "Resize layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "resize_h_ratio_list",
|
|
"type": "float32[]",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "resize_w_ratio_list",
|
|
"type": "float32[]",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "resize_f_ratio_list",
|
|
"type": "float32[]",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "method",
|
|
"type": "string",
|
|
"description": "Resize method, either bilinear or nearest_neighbor",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "resize_bilinear_pixels_mode",
|
|
"type": "string",
|
|
"description": "Bilinear resize variant, either half_pixels, align_corners, or disabled (where both align_corners and half_pixels are false)",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "shortcut",
|
|
"description": "Shortcut layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "slice",
|
|
"description": "Slice layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "height_slice",
|
|
"type": "int64[]",
|
|
"visible": false,
|
|
"description": "Slice in the height axis (start, stop, step)"
|
|
},
|
|
{
|
|
"name": "width_slice",
|
|
"type": "int64[]",
|
|
"visible": false,
|
|
"description": "Slice in the width axis (start, stop, step)"
|
|
},
|
|
{
|
|
"name": "features_slice",
|
|
"type": "int64[]",
|
|
"visible": false,
|
|
"description": "Slice in the features axis (start, stop, step)"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "softmax",
|
|
"category": "Activation",
|
|
"description": "Softmax layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "groups",
|
|
"type": "int64",
|
|
"description": "Number of groups input channels and output channels are divided into",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "space_to_depth",
|
|
"description": "Space to depth layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "block_sizes",
|
|
"type": "int64[]",
|
|
"description": "Block size along each spatial axis",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "space_to_depth_type",
|
|
"type": "string",
|
|
"description": "Space to depth variant, either classic_dcr (depth-column-row) classic_crd (column-row-depth), serial (used by Transformers patchify function), or focus (Yolov5-like)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "spatial_flatten_output",
|
|
"type": "boolean",
|
|
"description": "Whether to flatten the layer's output to one row",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "output_mux",
|
|
"description": "Output muxer layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "concat",
|
|
"category": "Tensor",
|
|
"description": "Concatenation layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "concat_axis",
|
|
"type": "int64",
|
|
"description": "Axis to concatenate along, either features or spatial_w (which means the width axis)",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "spatial_w_concat",
|
|
"type": "boolean",
|
|
"description": "Whether the concat operation is in the width dimension",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "matmul",
|
|
"description": "Matrix multiplication layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "kernel_shape",
|
|
"type": "int64[]",
|
|
"label": "kernel",
|
|
"description": "Shape of the kernel in Tensorflow convention (kernel height, kernel width, features in, features out)",
|
|
"visible": true
|
|
},
|
|
{
|
|
"name": "dynamic_weights",
|
|
"type": "boolean",
|
|
"description": "Whether the layer's weights are data driven",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "transpose_matmul_input",
|
|
"type": "boolean",
|
|
"description": "Whether to transpose the width and the features axes of the layer's second input tensor",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ew_add",
|
|
"description": "Elementwise addition layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ew_div",
|
|
"description": "Elementwise division layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ew_mult",
|
|
"description": "Elementwise multiplication layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "ew_sub",
|
|
"description": "Elementwise subtraction layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
},
|
|
{
|
|
"name": "activation",
|
|
"type": "string",
|
|
"description": "Activation function name",
|
|
"visible": false
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "demux",
|
|
"description": "Demuxer layer",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "row_splitter",
|
|
"description": "Splits the input tensor along the height axis",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "feature_splitter",
|
|
"description": "Splits the input tensor along the features axis",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "const_input",
|
|
"category": "Constant",
|
|
"description": "Constant input",
|
|
"attributes": [
|
|
{
|
|
"name": "original_names",
|
|
"type": "string[]",
|
|
"description": "Name of this layer in the original framework, such as Pytorch or Tensorflow",
|
|
"visible": false
|
|
}]
|
|
},
|
|
{
|
|
"name": "inv_pos",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "exp",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "silu",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "leaky",
|
|
"category": "Activation"
|
|
},
|
|
{
|
|
"name": "layer_normalization",
|
|
"category": "Normalization"
|
|
}
|
|
]
|
|
|