wehub-resource-sync
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
1219 lines
36 KiB
JSON
1219 lines
36 KiB
JSON
[
|
|
{
|
|
"name": "Add",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "pot_scale_int16", "type": "boolean", "default": true }
|
|
],
|
|
"inputs": [
|
|
{ "name": "A", "type": "T" },
|
|
{ "name": "B", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "C", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ArgMax",
|
|
"attributes": [
|
|
{ "name": "output_type", "type": "TensorType", "default": "FLOAT32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "dimension", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ArgMin",
|
|
"attributes": [
|
|
{ "name": "output_type", "type": "TensorType", "default": "FLOAT32" }
|
|
]
|
|
},
|
|
{
|
|
"name": "AveragePool2D",
|
|
"category": "Pool",
|
|
"attributes": [
|
|
{ "name": "padding", "type": "Padding", "default": "SAME" },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "stride_w", "type": "int32" },
|
|
{ "name": "stride_h", "type": "int32" },
|
|
{ "name": "filter_width", "type": "int32" },
|
|
{ "name": "filter_height", "type": "int32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "BatchMatMul",
|
|
"attributes": [
|
|
{ "name": "adjoint_lhs", "type": "boolean", "default": false },
|
|
{ "name": "adjoint_rhs", "type": "boolean", "default": false },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "BidirectionalSequenceLSTM",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "cell_clip", "type": "float32", "default": 0 },
|
|
{ "name": "proj_clip", "type": "float32", "default": 0 },
|
|
{ "name": "merge_outputs", "type": "boolean", "default": false },
|
|
{ "name": "time_major", "type": "boolean", "default": true },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "BidirectionalSequenceRNN",
|
|
"attributes": [
|
|
{ "name": "time_major", "type": "boolean", "default": false },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "merge_outputs", "type": "boolean", "default": false },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Bucketize",
|
|
"attributes": [
|
|
{ "name": "boundaries", "type": "float32[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "Call",
|
|
"attributes": [
|
|
{ "name": "subgraph", "type": "uint32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "CallOnce",
|
|
"attributes": [
|
|
{ "name": "init_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "Cast",
|
|
"attributes": [
|
|
{ "name": "in_data_type", "type": "TensorType", "default": "FLOAT32" },
|
|
{ "name": "out_data_type", "type": "TensorType", "default": "FLOAT32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ConcatEmbeddings",
|
|
"attributes": [
|
|
{ "name": "num_channels", "type": "int32", "default": 0 },
|
|
{ "name": "num_columns_per_channel", "type": "int32[]", "default": 0 },
|
|
{ "name": "embedding_dim_per_channel", "type": "int32[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "Concatenation",
|
|
"category": "Tensor",
|
|
"attributes": [
|
|
{ "name": "axis", "type": "int32", "default": 0 },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "inputs", "list": true }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Conv2D",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "padding", "type": "Padding", "default": "SAME", "description": "`SAME`|`VALID`" },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE", "description": "`NONE`|`RELU`|`RELU6`" },
|
|
{ "name": "stride_w", "type": "int32", "default": 0, "description": "stride of the filter window" },
|
|
{ "name": "stride_h", "type": "int32", "default": 0, "description": "stride of the filter window" },
|
|
{ "name": "dilation_w_factor", "type": "int32", "default": 1 },
|
|
{ "name": "dilation_h_factor", "type": "int32", "default": 1 },
|
|
{ "name": "quantized_bias_type", "type": "TensorType", "default": "FLOAT32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "description": "4D tensor" },
|
|
{ "name": "filter" },
|
|
{ "name": "bias", "description": "(optional)" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "description": "result of 2D convolution of the input tensor" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Cumsum",
|
|
"attributes": [
|
|
{ "name": "exclusive", "type": "boolean", "default": false },
|
|
{ "name": "reverse", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Densify",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "DepthToSpace",
|
|
"attributes": [
|
|
{ "name": "block_size", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "DepthwiseConv2D",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "padding", "type": "Padding", "default": "SAME" },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "stride_w", "type": "int32", "default": 0 },
|
|
{ "name": "stride_h", "type": "int32", "default": 0 },
|
|
{ "name": "depth_multiplier", "type": "int32", "default": 0 },
|
|
{ "name": "dilation_w_factor", "type": "int32", "default": 1 },
|
|
{ "name": "dilation_h_factor", "type": "int32", "default": 1 }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "weights", "type": "T" },
|
|
{ "name": "bias", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Dequantize",
|
|
"category": "Quantization",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Div",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "A", "type": "T" },
|
|
{ "name": "B", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "C", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "EmbeddingLookupSparse",
|
|
"attributes": [
|
|
{ "name": "combiner", "type": "CombinerType", "default": "SUM" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Exp",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ExpandDims",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "axis_param" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ExtractFeatures",
|
|
"inputs": [
|
|
{ "name": "ngrams" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "features" },
|
|
{ "name": "weights" }
|
|
]
|
|
},
|
|
{
|
|
"name": "FakeQuant",
|
|
"attributes": [
|
|
{ "name": "min", "type": "float32", "default": 0 },
|
|
{ "name": "max", "type": "float32", "default": 0 },
|
|
{ "name": "num_bits", "type": "int32", "default": 0 },
|
|
{ "name": "narrow_range", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Fill",
|
|
"inputs": [
|
|
{ "name": "dims", "type": "T" },
|
|
{ "name": "value", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "FullyConnected",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "weights_format", "type": "FullyConnectedOptionsWeightsFormat", "default": "DEFAULT" },
|
|
{ "name": "keep_num_dims", "type": "boolean", "default": false },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false },
|
|
{ "name": "quantized_bias_type", "type": "TensorType", "default": "FLOAT32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "weights", "type": "T" },
|
|
{ "name": "bias", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Gather",
|
|
"category": "Transform",
|
|
"attributes": [
|
|
{ "name": "axis", "default": 0, "type": "int32" },
|
|
{ "name": "batch_dims", "type": "int32", "default": 0 }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "positions" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Gelu",
|
|
"category": "Activation",
|
|
"attributes": [
|
|
{ "name": "approximate", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "HardSwish",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Hashtable",
|
|
"attributes": [
|
|
{ "name": "table_id", "type": "int32", "default": 0 },
|
|
{ "name": "key_dtype", "type": "TensorType", "default": "FLOAT32" },
|
|
{ "name": "value_dtype", "type": "TensorType", "default": "FLOAT32" }
|
|
]
|
|
},
|
|
{
|
|
"name": "HashtableLookup",
|
|
"inputs": [
|
|
{ "name": "key" },
|
|
{ "name": "keys" },
|
|
{ "name": "values" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "value" },
|
|
{ "name": "hits" }
|
|
]
|
|
},
|
|
{
|
|
"name": "If",
|
|
"attributes": [
|
|
{ "name": "then_subgraph_index", "type": "int32", "default": 0 },
|
|
{ "name": "else_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "InstanceNorm",
|
|
"attributes": [
|
|
{ "name": "epsilon", "type": "float32", "default": 0 },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
|
|
]
|
|
},
|
|
{
|
|
"name": "LeakyRelu",
|
|
"attributes": [
|
|
{ "name": "alpha", "type": "float32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "LocalResponseNormalization",
|
|
"category": "Normalization",
|
|
"attributes": [
|
|
{ "name": "radius", "type": "int32", "default": 0 },
|
|
{ "name": "bias", "type": "float32", "default": 0 },
|
|
{ "name": "alpha", "type": "float32", "default": 0 },
|
|
{ "name": "beta", "type": "float32", "default": 0 }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "LogicalOr",
|
|
"inputs": [
|
|
{ "name": "A", "type": "T" },
|
|
{ "name": "B", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "C", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Logistic",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "LogSoftmax",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "LSHProjection",
|
|
"attributes": [
|
|
{ "name": "type", "type": "LSHProjectionType", "default": "UNKNOWN" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "hash" },
|
|
{ "name": "input" },
|
|
{ "name": "weight" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "LSTM",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "cell_clip", "type": "float32", "default": 0 },
|
|
{ "name": "proj_clip", "type": "float32", "default": 0 },
|
|
{ "name": "kernel_type", "type": "LSTMKernelType", "default": "FULL" },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T", "description": "Input tensor." },
|
|
{ "name": "input_input_weights", "type": "T", "option": "optional", "description": "Input to input weights tensor.", "visible": false },
|
|
{ "name": "input_forget_weights", "type": "T", "description": "Input to forget weights tensor.", "visible": false },
|
|
{ "name": "input_cell_weights", "type": "T", "description": "Input to cell weights tensor.", "visible": false },
|
|
{ "name": "input_output_weights", "type": "T", "description": "Input to output weights tensor.", "visible": false },
|
|
{ "name": "recurrent_input_weights", "type": "T", "option": "optional", "description": "Recurrent to input weights tensor.", "visible": false },
|
|
{ "name": "recurrent_forget_weights", "type": "T", "description": "Recurrent to forget weights tensor.", "visible": false },
|
|
{ "name": "recurrent_cell_weights", "type": "T", "description": "Recurrent to cell weights tensor.", "visible": false },
|
|
{ "name": "recurrent_output_weights", "type": "T", "description": "Recurrent to output weights tensor.", "visible": false },
|
|
{ "name": "cell_input_weights", "type": "T", "option": "optional", "description": "Cell to input weights tensor.", "visible": false },
|
|
{ "name": "cell_forget_weights", "type": "T", "option": "optional", "description": "Cell to forget weights tensor.", "visible": false },
|
|
{ "name": "cell_output_weights", "type": "T", "option": "optional", "description": "Cell to output weights tensor.", "visible": false },
|
|
{ "name": "input_gate_bias", "type": "T", "option": "optional", "description": "Input gate bias tensor.", "visible": false },
|
|
{ "name": "forget_gate_bias", "type": "T", "description": "Forget gate bias tensor.", "visible": false },
|
|
{ "name": "cell_gate_bias", "type": "T", "description": "Cell gate bias tensor.", "visible": false },
|
|
{ "name": "output_gate_bias", "type": "T", "description": "Output gate bias tensor.", "visible": false },
|
|
{ "name": "projection_weights", "type": "T", "option": "optional", "description": "Projection weights tensor.", "visible": false },
|
|
{ "name": "projection_bias", "type": "T", "option": "optional", "description": "Projection bias tensor.", "visible": false }
|
|
],
|
|
"outputs": [
|
|
{ "name": "scratch", "type": "T" },
|
|
{ "name": "output_state", "type": "T" },
|
|
{ "name": "cell_state", "type": "T" },
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Maximum",
|
|
"inputs": [
|
|
{ "name": "input1" },
|
|
{ "name": "input2" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "MaxPool2D",
|
|
"category": "Pool",
|
|
"attributes": [
|
|
{ "name": "padding", "type": "Padding", "default": "SAME" },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "stride_w", "type": "int32" },
|
|
{ "name": "stride_h", "type": "int32" },
|
|
{ "name": "filter_width", "type": "int32" },
|
|
{ "name": "filter_height", "type": "int32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Mean",
|
|
"attributes": [
|
|
{ "name": "keep_dims", "type": "boolean" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "axis", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Minimum",
|
|
"inputs": [
|
|
{ "name": "input1" },
|
|
{ "name": "input2" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "MirrorPad",
|
|
"attributes": [
|
|
{ "name": "mode", "type": "MirrorPadMode", "default": "REFLECT" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Mul",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "A", "type": "T" },
|
|
{ "name": "B", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "C", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Normalize",
|
|
"category": "Normalization",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "NotEqual",
|
|
"inputs": [
|
|
{ "name": "A", "type": "T" },
|
|
{ "name": "B", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "C" }
|
|
]
|
|
},
|
|
{
|
|
"name": "OneHot",
|
|
"attributes": [
|
|
{ "name": "axis", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "Pack",
|
|
"attributes": [
|
|
{ "name": "values_count", "type": "int32", "default": 0 },
|
|
{ "name": "axis", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "Pad",
|
|
"category": "Tensor",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "paddings" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Predict",
|
|
"inputs": [
|
|
{ "name": "hashes" },
|
|
{ "name": "keys" },
|
|
{ "name": "labels" },
|
|
{ "name": "weights" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "label" },
|
|
{ "name": "weight" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Prelu",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "slope", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Quantize",
|
|
"category": "Quantization",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Range",
|
|
"attributes": [
|
|
{ "name": "keep_dims", "type": "boolean" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "start", "type": "T" },
|
|
{ "name": "limit", "type": "T" },
|
|
{ "name": "delta", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ReduceMax",
|
|
"attributes": [
|
|
{ "name": "keep_dims", "type": "boolean" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "axis", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ReduceMin",
|
|
"attributes": [
|
|
{ "name": "keep_dims", "type": "boolean" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "axis", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ReduceWindow",
|
|
"attributes": [
|
|
{ "name": "reduce_function", "type": "ReduceWindowFunction", "default": "UNSUPPORTED" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Relu",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Relu6",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Reshape",
|
|
"category": "Shape",
|
|
"attributes": [
|
|
{ "name": "new_shape", "type": "int32[]", "default": 0 }
|
|
],
|
|
"inputs": [
|
|
{ "name": "data", "type": "T" },
|
|
{ "name": "shape", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "reshaped", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ResizeBilinear",
|
|
"attributes": [
|
|
{ "name": "align_corners", "default": false, "type": "boolean" },
|
|
{ "name": "new_height", "type": "int32", "default": 0 },
|
|
{ "name": "new_width", "type": "int32", "default": 0 },
|
|
{ "name": "half_pixel_centers", "type": "boolean", "default": false }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "size" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "ResizeNearestNeighbor",
|
|
"attributes": [
|
|
{ "name": "align_corners", "type": "boolean", "default": false },
|
|
{ "name": "half_pixel_centers", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "ReverseSequence",
|
|
"attributes": [
|
|
{ "name": "seq_dim", "type": "int32", "default": 0 },
|
|
{ "name": "batch_dim", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "RmsNorm",
|
|
"attributes": [
|
|
{ "name": "epsilon", "type": "float32", "default": 0 },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" }
|
|
]
|
|
},
|
|
{
|
|
"name": "RNN",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
|
|
],
|
|
"inputs": [
|
|
{ "name": "X", "type": "T" },
|
|
{ "name": "W", "type": "T" },
|
|
{ "name": "R", "type": "T" },
|
|
{ "name": "b", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "hidden", "type": "T" },
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Rsqrt",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "RunModel",
|
|
"attributes": [
|
|
{ "name": "location", "type": "string", "default": null },
|
|
{ "name": "signature", "type": "string", "default": null }
|
|
]
|
|
},
|
|
{
|
|
"name": "Shape",
|
|
"attributes": [
|
|
{ "name": "out_type", "type": "TensorType", "default": "FLOAT32" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Sin",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "SkipGram",
|
|
"inputs": [
|
|
{ "name": "inputs" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "ngrams" }
|
|
],
|
|
"attributes": [
|
|
{ "name": "ngram_size", "type": "int32", "default": 0 },
|
|
{ "name": "max_skip_size", "type": "int32", "default": 0 },
|
|
{ "name": "include_all_ngrams", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Slice",
|
|
"category": "Tensor",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "begin" },
|
|
{ "name": "size" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Softmax",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
],
|
|
"attributes": [
|
|
{ "name": "beta", "type": "float32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "SpaceToDepth",
|
|
"attributes": [
|
|
{ "name": "block_size", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "SparseToDense",
|
|
"attributes": [
|
|
{ "name": "validate_indices", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Split",
|
|
"category": "Tensor",
|
|
"inputs": [
|
|
{ "name": "axis" },
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
],
|
|
"attributes": [
|
|
{ "name": "num_splits", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "SplitV",
|
|
"attributes": [
|
|
{ "name": "num_splits", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "Squeeze",
|
|
"category": "Transform",
|
|
"inputs": [
|
|
{ "name": "input" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
],
|
|
"attributes": [
|
|
{ "name": "squeeze_dims", "type": "int32[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloBroadcastInDim",
|
|
"attributes": [
|
|
{ "name": "broadcast_dimensions", "type": "int64[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloCase",
|
|
"attributes": [
|
|
{ "name": "branch_subgraph_indices", "type": "int32[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloCompare",
|
|
"attributes": [
|
|
{ "name": "comparison_direction", "type": "StablehloComparisonDirection", "default": "STABLEHLO_COMPARISON_DIRECTION_EQ" },
|
|
{ "name": "compare_type", "type": "StablehloComparisonType", "default": "STABLEHLO_COMPARISON_TYPE_NOTYPE" }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloConcatenate",
|
|
"attributes": [
|
|
{ "name": "dimension", "type": "int64", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloConvolution",
|
|
"attributes": [
|
|
{ "name": "window_strides", "type": "int64[]", "default": 0 },
|
|
{ "name": "padding", "type": "int64[]", "default": 0 },
|
|
{ "name": "lhs_dilation", "type": "int64[]", "default": 0 },
|
|
{ "name": "rhs_dilation", "type": "int64[]", "default": 0 },
|
|
{ "name": "window_reversal", "type": "boolean", "default": false },
|
|
{ "name": "input_batch_dimension", "type": "int64", "default": 0 },
|
|
{ "name": "input_feature_dimension", "type": "int64", "default": 0 },
|
|
{ "name": "input_spatial_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "kernel_input_feature_dimension", "type": "int64", "default": 0 },
|
|
{ "name": "kernel_output_feature_dimension", "type": "int64", "default": 0 },
|
|
{ "name": "kernel_spatial_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "output_batch_dimension", "type": "int64", "default": 0 },
|
|
{ "name": "output_feature_dimension", "type": "int64", "default": 0 },
|
|
{ "name": "output_spatial_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "feature_group_count", "type": "int64", "default": 0 },
|
|
{ "name": "batch_group_count", "type": "int64", "default": 0 },
|
|
{ "name": "precision_config", "type": "StablehloPrecisionConfig[]", "default": "DEFAULT" }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloCustomCall",
|
|
"attributes": [
|
|
{ "name": "call_target_name", "type": "string", "default": null },
|
|
{ "name": "has_side_effect", "type": "boolean", "default": false },
|
|
{ "name": "backend_config", "type": "string", "default": null },
|
|
{ "name": "api_version", "type": "int32", "default": 0 },
|
|
{ "name": "called_computations", "type": "int32[]", "default": 0 },
|
|
{ "name": "custom_attributes", "type": "uint8[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloDotGeneral",
|
|
"attributes": [
|
|
{ "name": "lhs_batching_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "rhs_batching_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "lhs_contracting_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "rhs_contracting_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "precision_config", "type": "StablehloPrecisionConfig[]", "default": "DEFAULT" }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloDynamicSlice",
|
|
"attributes": [
|
|
{ "name": "slice_sizes", "type": "int64[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloGather",
|
|
"attributes": [
|
|
{ "name": "offset_dims", "type": "int64[]", "default": 0 },
|
|
{ "name": "collapsed_slice_dims", "type": "int64[]", "default": 0 },
|
|
{ "name": "start_index_map", "type": "int64[]", "default": 0 },
|
|
{ "name": "index_vector_dim", "type": "int64", "default": 0 },
|
|
{ "name": "slice_sizes", "type": "int64[]", "default": 0 },
|
|
{ "name": "indices_are_sorted", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloIota",
|
|
"attributes": [
|
|
{ "name": "iota_dimension", "type": "int64", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloPad",
|
|
"attributes": [
|
|
{ "name": "edge_padding_low", "type": "int64[]", "default": 0 },
|
|
{ "name": "edge_padding_high", "type": "int64[]", "default": 0 },
|
|
{ "name": "interior_padding", "type": "int64[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloReduce",
|
|
"attributes": [
|
|
{ "name": "dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "body_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloReduceWindow",
|
|
"attributes": [
|
|
{ "name": "window_dimensions", "type": "int64[]", "default": 0 },
|
|
{ "name": "window_strides", "type": "int64[]", "default": 0 },
|
|
{ "name": "base_dilations", "type": "int64[]", "default": 0 },
|
|
{ "name": "window_dilations", "type": "int64[]", "default": 0 },
|
|
{ "name": "padding", "type": "int64[]", "default": 0 },
|
|
{ "name": "body_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloRngBitGenerator",
|
|
"attributes": [
|
|
{ "name": "algorithm", "type": "RngAlgorithm", "default": "DEFAULT" }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloScatter",
|
|
"attributes": [
|
|
{ "name": "indices_are_sorted", "type": "boolean", "default": false },
|
|
{ "name": "update_window_dims", "type": "int64[]", "default": 0 },
|
|
{ "name": "inserted_window_dims", "type": "int64[]", "default": 0 },
|
|
{ "name": "scatter_dims_to_operand_dims", "type": "int64[]", "default": 0 },
|
|
{ "name": "index_vector_dim", "type": "int64", "default": 0 },
|
|
{ "name": "unique_indices", "type": "boolean", "default": false },
|
|
{ "name": "update_computation_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloSlice",
|
|
"attributes": [
|
|
{ "name": "start_indices", "type": "int64[]", "default": 0 },
|
|
{ "name": "limit_indices", "type": "int64[]", "default": 0 },
|
|
{ "name": "strides", "type": "int64[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloSort",
|
|
"attributes": [
|
|
{ "name": "dimension", "type": "int64", "default": 0 },
|
|
{ "name": "is_stable", "type": "boolean", "default": false },
|
|
{ "name": "comparator_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloTranspose",
|
|
"attributes": [
|
|
{ "name": "permutation", "type": "int64[]", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StablehloWhile",
|
|
"attributes": [
|
|
{ "name": "cond_subgraph_index", "type": "int32", "default": 0 },
|
|
{ "name": "body_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "StridedSlice",
|
|
"category": "Tensor",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "begin" },
|
|
{ "name": "end" },
|
|
{ "name": "strides" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
],
|
|
"attributes": [
|
|
{ "name": "begin_mask", "type": "int32", "default": 0 },
|
|
{ "name": "end_mask", "type": "int32", "default": 0 },
|
|
{ "name": "ellipsis_mask", "type": "int32", "default": 0 },
|
|
{ "name": "new_axis_mask", "type": "int32", "default": 0 },
|
|
{ "name": "shrink_axis_mask", "type": "int32", "default": 0 },
|
|
{ "name": "offset", "type": "boolean", "default": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Sub",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "pot_scale_int16", "type": "boolean", "default": true }
|
|
],
|
|
"inputs": [
|
|
{ "name": "A", "type": "T" },
|
|
{ "name": "B", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "C", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Sum",
|
|
"attributes": [
|
|
{ "name": "keep_dims", "type": "boolean" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "axis", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "SVDF",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "rank", "type": "int32", "default": 0 },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" },
|
|
{ "name": "feature", "type": "T" },
|
|
{ "name": "time", "type": "T" },
|
|
{ "name": "bias", "type": "T" },
|
|
{ "name": "state", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "state_out", "type": "T" },
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Tanh",
|
|
"category": "Activation",
|
|
"inputs": [
|
|
{ "name": "input", "type": "T" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output", "type": "T" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Tile",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "multipliers" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Transpose",
|
|
"category": "Transform",
|
|
"inputs": [
|
|
{ "name": "input" },
|
|
{ "name": "perm" }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "TransposeConv",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "padding", "type": "Padding", "default": "SAME" },
|
|
{ "name": "stride_w", "type": "int32", "default": 0 },
|
|
{ "name": "stride_h", "type": "int32", "default": 0 },
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "quantized_bias_type", "type": "TensorType", "default": "FLOAT32" }
|
|
],
|
|
"inputs": [
|
|
{ "name": "output_shape" },
|
|
{ "name": "weights" },
|
|
{ "name": "input" },
|
|
{ "name": "bias", "optional": true }
|
|
],
|
|
"outputs": [
|
|
{ "name": "output" }
|
|
]
|
|
},
|
|
{
|
|
"name": "UnidirectionalSequenceLSTM",
|
|
"category": "Layer",
|
|
"attributes": [
|
|
{ "name": "fused_activation_function", "type": "ActivationFunctionType", "default": "NONE" },
|
|
{ "name": "cell_clip", "type": "float32", "default": 0 },
|
|
{ "name": "proj_clip", "type": "float32", "default": 0 },
|
|
{ "name": "time_major", "type": "boolean", "default": false },
|
|
{ "name": "asymmetric_quantize_inputs", "type": "boolean", "default": false },
|
|
{ "name": "diagonal_recurrent_tensors", "type": "boolean", "default": false }
|
|
],
|
|
"inputs": [
|
|
{ "name": "input", "type": "T", "description": "Input tensor." },
|
|
{ "name": "input_input_weights", "type": "T", "option": "optional", "description": "Input to input weights tensor.", "visible": false },
|
|
{ "name": "input_forget_weights", "type": "T", "description": "Input to forget weights tensor.", "visible": false },
|
|
{ "name": "input_cell_weights", "type": "T", "description": "Input to cell weights tensor.", "visible": false },
|
|
{ "name": "input_output_weights", "type": "T", "description": "Input to output weights tensor.", "visible": false },
|
|
{ "name": "recurrent_input_weights", "type": "T", "option": "optional", "description": "Recurrent to input weights tensor.", "visible": false },
|
|
{ "name": "recurrent_forget_weights", "type": "T", "description": "Recurrent to forget weights tensor.", "visible": false },
|
|
{ "name": "recurrent_cell_weights", "type": "T", "description": "Recurrent to cell weights tensor.", "visible": false },
|
|
{ "name": "recurrent_output_weights", "type": "T", "description": "Recurrent to output weights tensor.", "visible": false },
|
|
{ "name": "cell_input_weights", "type": "T", "option": "optional", "description": "Cell to input weights tensor.", "visible": false },
|
|
{ "name": "cell_forget_weights", "type": "T", "option": "optional", "description": "Cell to forget weights tensor.", "visible": false },
|
|
{ "name": "cell_output_weights", "type": "T", "option": "optional", "description": "Cell to output weights tensor.", "visible": false },
|
|
{ "name": "input_gate_bias", "type": "T", "option": "optional", "description": "Input gate bias tensor.", "visible": false },
|
|
{ "name": "forget_gate_bias", "type": "T", "description": "Forget gate bias tensor.", "visible": false },
|
|
{ "name": "cell_gate_bias", "type": "T", "description": "Cell gate bias tensor.", "visible": false },
|
|
{ "name": "output_gate_bias", "type": "T", "description": "Output gate bias tensor.", "visible": false },
|
|
{ "name": "projection_weights", "type": "T", "option": "optional", "description": "Projection weights tensor.", "visible": false },
|
|
{ "name": "projection_bias", "type": "T", "option": "optional", "description": "Projection bias tensor.", "visible": false },
|
|
{ "name": "output_state_in", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "cell_state_in", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "activation", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "cell_clip", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "proj_clip", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "time_major", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "input_layer_norm_weights", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "forget_layer_norm_weights", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "cell_layer_norm_weights", "type": "T", "option": "optional", "visible": false },
|
|
{ "name": "output_layer_norm_weights", "type": "T", "option": "optional", "visible": false }
|
|
]
|
|
},
|
|
{
|
|
"name": "Unique",
|
|
"attributes": [
|
|
{ "name": "idx_out_type", "type": "TensorType", "default": "INT32" }
|
|
]
|
|
},
|
|
{
|
|
"name": "Unpack",
|
|
"attributes": [
|
|
{ "name": "num", "type": "int32", "default": 0 },
|
|
{ "name": "axis", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "UnsortedSegmentProd",
|
|
"attributes": [
|
|
{ "name": "num_segments", "type": "int32", "default": 0 }
|
|
]
|
|
},
|
|
{
|
|
"name": "VarHandle",
|
|
"attributes": [
|
|
{ "name": "container", "type": "string", "default": null },
|
|
{ "name": "shared_name", "type": "string", "default": null }
|
|
]
|
|
},
|
|
{
|
|
"name": "While",
|
|
"attributes": [
|
|
{ "name": "cond_subgraph_index", "type": "int32", "default": 0 },
|
|
{ "name": "body_subgraph_index", "type": "int32", "default": 0 }
|
|
]
|
|
}
|
|
] |