[
  {
    "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": "adj_x", "type": "boolean", "default": false },
      { "name": "adj_y", "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",
    "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",
    "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": "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": "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": "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 }
    ]
  }
]