| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218 |
- [
- {
- "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",
- "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",
- "category": "Activation",
- "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": "Padv2",
- "category": "Tensor",
- "inputs": [
- { "name": "input" },
- { "name": "paddings" },
- { "name": "constant_values" }
- ],
- "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": "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": "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 }
- ]
- }
- ]
|