[ { "name": "torch.nn.modules.conv.Conv1d", "category": "Layer", "attributes": [ { "name": "output_padding", "visible": false }, { "name": "in_channels", "visible": false }, { "name": "out_channels", "visible": false }, { "name": "groups", "default": 1 }, { "name": "transposed", "default": false }, { "name": "padding", "default": [ 0 ] }, { "name": "dilation", "default": [ 1 ] }, { "name": "stride", "default": [ 1 ] } ] }, { "name": "torch.nn.modules.conv.ConvTranspose1d", "category": "Layer", "attributes": [ { "name": "output_padding", "visible": false }, { "name": "in_channels", "visible": false }, { "name": "out_channels", "visible": false }, { "name": "groups", "default": 1 }, { "name": "transposed", "default": true }, { "name": "padding", "default": [ 0 ] }, { "name": "dilation", "default": [ 1 ] }, { "name": "stride", "default": [ 1 ] } ] }, { "name": "torch.nn.modules.conv.Conv2d", "category": "Layer", "attributes": [ { "name": "output_padding", "visible": false }, { "name": "in_channels", "visible": false }, { "name": "out_channels", "visible": false }, { "name": "groups", "default": 1 }, { "name": "transposed", "default": false }, { "name": "padding", "default": [ 0, 0 ] }, { "name": "dilation", "default": [ 1, 1 ] }, { "name": "stride", "default": [ 1, 1 ] } ] }, { "name": "torch.nn.modules.conv.ConvTranspose2d", "category": "Layer", "attributes": [ { "name": "output_padding", "visible": false }, { "name": "in_channels", "visible": false }, { "name": "out_channels", "visible": false }, { "name": "groups", "default": 1 }, { "name": "transposed", "default": true }, { "name": "padding", "default": [ 0, 0 ] }, { "name": "dilation", "default": [ 1, 1 ] }, { "name": "stride", "default": [ 1, 1 ] } ] }, { "name": "torch.nn.modules.conv.Conv3d", "category": "Layer" }, { "name": "torch.nn.modules.conv.ConvTranspose3d", "category": "Layer" }, { "name": "torch.nn.modules.linear.Linear", "category": "Layer", "attributes": [] }, { "name": "torch.nn.modules.activation.Sigmoid", "category": "Activation" }, { "name": "torch.nn.modules.activation.Softmax", "category": "Activation" }, { "name": "torch.nn.modules.activation.Softmax2d", "category": "Activation" }, { "name": "torch.nn.modules.activation.Softplus", "category": "Activation" }, { "name": "torch.nn.modules.activation.Tanh", "category": "Activation" }, { "name": "torch.nn.modules.activation.LogSoftmax", "category": "Activation" }, { "name": "torch.nn.modules.activation.ReLU", "category": "Activation", "attributes": [ { "name": "inplace", "default": false, "visible": false }, { "name": "threshold", "default": 0 }, { "name": "value", "default": 0 } ] }, { "name": "torch.nn.modules.activation.ReLU6", "category": "Activation" }, { "name": "torch.nn.modules.activation.PReLU", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.ELU", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.GELU", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.GLU", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.Hardtanh", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.Hardswish", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.Hardsigmoid", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.activation.LeakyReLU", "category": "Activation", "attributes": [] }, { "name": "torch.nn.modules.pooling.MaxPool1d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.MaxPool2d", "category": "Pool", "attributes": [ { "name": "padding", "default": 0 }, { "name": "dilation", "default": 1 }, { "name": "return_indices", "default": false }, { "name": "ceil_mode", "visible": false } ] }, { "name": "torch.nn.modules.pooling.MaxPool3d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.MaxUnpool1d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.MaxUnpool2d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.MaxUnpool3d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AvgPool2d", "category": "Pool", "attributes": [ { "name": "padding", "default": 0 }, { "name": "count_include_pad", "default": true }, { "name": "ceil_mode", "visible": false } ] }, { "name": "torch.nn.modules.pooling.AvgPool3d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AdaptiveAvgPool1d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AdaptiveAvgPool2d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AdaptiveAvgPool3d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AdaptiveMaxPool1d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AdaptiveMaxPool2d", "category": "Pool" }, { "name": "torch.nn.modules.pooling.AdaptiveMaxPool3d", "category": "Pool" }, { "name": "torch.nn.modules.batchnorm.BatchNorm1d", "category": "Normalization", "attributes": [] }, { "name": "torch.nn.modules.batchnorm.BatchNorm2d", "category": "Normalization", "attributes": [ { "name": "eps", "default": 0.00001 }, { "name": "momentum", "default": 0.1 }, { "name": "affine", "default": true }, { "name": "track_running_stats", "default": true } ], "inputs": [ { "name": "input" }, { "name": "weight" }, { "name": "bias" }, { "name": "running_mean" }, { "name": "running_var" }, { "name": "num_batches_tracked", "visible": false } ] }, { "name": "torch.nn.modules.normalization.GroupNorm", "category": "Normalization" }, { "name": "torch.nn.modules.normalization.LayerNorm", "category": "Normalization" }, { "name": "torch.nn.modules.dropout.Dropout2d", "category": "Dropout", "attributes": [ { "name": "inplace", "default": false, "visible": false }, { "name": "p", "default": 0.5 } ] }, { "name": "torch.nn.modules.dropout.Dropout", "category": "Dropout", "attributes": [ { "name": "inplace", "default": false, "visible": false }, { "name": "p", "default": 0.5 } ] }, { "name": "torch.nn.modules.upsampling.Upsample", "category": "Data" }, { "name": "torch.nn.modules.rnn.GRU", "category": "Layer" }, { "name": "torch.nn.modules.rnn.GRUCell", "category": "Layer" }, { "name": "torch.nn.modules.rnn.LSTM", "category": "Layer", "attributes": [ { "name": "dropout", "default": 0 }, { "name": "dropout_state", "default": {} }, { "name": "num_layers", "default": 1 }, { "name": "batch_first", "visible": false }, { "name": "bidirectional", "visible": false }, { "name": "bias", "visible": false } ], "inputs": [ { "name": "input" }, { "name": "weight_ih_l0", "visible": false }, { "name": "weight_hh_l0", "visible": false }, { "name": "bias_ih_l0", "visible": false }, { "name": "bias_hh_l0", "visible": false }, { "name": "weight_ih_l1", "visible": false }, { "name": "weight_hh_l1", "visible": false }, { "name": "bias_ih_l1", "visible": false }, { "name": "bias_hh_l1", "visible": false } ] }, { "name": "torch.nn.modules.rnn.LSTMCell", "category": "Layer" }, { "name": "torch.nn.modules.rnn.RNN", "category": "Layer" }, { "name": "torch.nn.modules.sparse.Embedding", "category": "Transform", "attributes": [ { "name": "norm_type", "default": 2 }, { "name": "scale_grad_by_freq", "default": false }, { "name": "sparse", "default": false }, { "name": "max_norm", "default": null }, { "name": "padding_idx", "default": null } ] }, { "name": "torch.nn.modules.padding.ReflectionPad1d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ReflectionPad2d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ReplicationPad1d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ReplicationPad2d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ReplicationPad3d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ZeroPad2d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ConstantPad1d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ConstantPad2d", "category": "Tensor" }, { "name": "torch.nn.modules.padding.ConstantPad3d", "category": "Tensor" }, { "name": "torch.nn.modules.pixelshuffle.PixelShuffle" }, { "name": "torch.nn.modules.instancenorm.InstanceNorm1d" }, { "name": "torch.nn.modules.instancenorm.InstanceNorm2d" }, { "name": "torch.nn.modules.instancenorm.InstanceNorm3d" }, { "name": "torch.nn.modules.normalization.CrossMapLRN2d", "category": "Normalization", "attributes": [ { "name": "alpha", "default": 0.0001 }, { "name": "beta", "default": 0.75 }, { "name": "k", "default": 1 } ] }, { "name": "torch._convolution:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "transposed", "type": "boolean", "default": false }, { "name": "output_padding", "type": "int64[]", "default": 0 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "benchmark", "type": "boolean", "visible": false }, { "name": "deterministic", "type": "boolean", "visible": false }, { "name": "cudnn_enabled", "type": "boolean", "visible": false }, { "name": "allow_tf32", "type": "boolean", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._convolution_mode:", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "string" }, { "name": "dilation", "type": "int64[]" }, { "name": "groups", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._convolution_nogroup:", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "string" }, { "name": "dilation", "type": "int64[]" }, { "name": "transposed", "type": "boolean" }, { "name": "output_padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._convolution:deprecated", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "transposed", "type": "boolean", "default": false }, { "name": "output_padding", "type": "int64[]", "default": 0 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "benchmark", "type": "boolean", "visible": false }, { "name": "deterministic", "type": "boolean", "visible": false }, { "name": "cudnn_enabled", "type": "boolean", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._convolution_mode:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "string" }, { "name": "dilation", "type": "int64[]" }, { "name": "groups", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv1d:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv1d:padding", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "string", "default": "valid" }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv2d:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv2d:padding", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "string", "default": "valid" }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv3d:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv3d:padding", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "string", "default": "valid" }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv_transpose1d:input", "category": "Layer", "attributes": [ { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "output_padding", "type": "int64[]", "default": 0 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "dilation", "type": "int64[]", "default": 1 } ], "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv_transpose2d:input", "category": "Layer", "attributes": [ { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "output_padding", "type": "int64[]", "default": 0 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "dilation", "type": "int64[]", "default": 1 } ], "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conv_transpose3d:input", "category": "Layer", "attributes": [ { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "output_padding", "type": "int64[]", "default": 0 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "dilation", "type": "int64[]", "default": 1 } ], "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.torchaudio.sox_effects_apply_effects_tensor", "inputs": [ { "name": "tensor", "type": "Tensor" }, { "name": "sample_rate", "type": "int64" }, { "name": "effects", "type": "string[][]" }, { "name": "channels_first", "type": "boolean", "default": true } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "ops.prepacked.conv2d_clamp_run", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "context": true }, { "name": "padding", "type": "int64[]", "context": true }, { "name": "dilation", "type": "int64[]", "context": true }, { "name": "groups", "type": "int64", "context": true }, { "name": "output_min", "type": "Scalar", "context": true }, { "name": "output_max", "type": "Scalar", "context": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.prepacked.linear_clamp_run", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "output_min", "type": "Scalar", "context": true }, { "name": "output_max", "type": "Scalar", "context": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.add", "inputs": [ { "name": "A", "type": "Tensor" }, { "name": "B", "type": "Tensor" }, { "name": "scale", "type": "float32" }, { "name": "zero_point", "type": "int64" } ], "outputs": [ { "name": "C", "type": "Tensor" } ] }, { "name": "ops.quantized.add_scalar:", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.add_scalar_relu:", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.add_scalar:Tensor", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Tensor" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.add_scalar_relu:Tensor", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Tensor" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.linear", "category": "Layer", "inputs": [ { "name": "X", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "Y_scale_i", "type": "float32" }, { "name": "Y_zero_point_i", "type": "int64" } ], "outputs": [ { "name": "Y", "type": "Tensor" } ] }, { "name": "ops.quantized.linear_relu", "category": "Layer", "inputs": [ { "name": "X", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "Y_scale_i", "type": "float32" }, { "name": "Y_zero_point_i", "type": "int64" } ], "outputs": [ { "name": "Y", "type": "Tensor" } ] }, { "name": "ops.quantized.linear_dynamic", "category": "Layer", "inputs": [ { "name": "X", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "reduce_range", "type": "boolean", "default": false } ], "outputs": [ { "name": "Y", "type": "Tensor" } ] }, { "name": "ops.quantized.linear_relu_dynamic", "category": "Layer", "inputs": [ { "name": "X", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "Y_scale_i", "type": "float32" }, { "name": "Y_zero_point_i", "type": "int64" } ], "outputs": [ { "name": "Y", "type": "Tensor" } ] }, { "name": "torch.dequantize", "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv1d:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1, "context": true }, { "name": "padding", "type": "int64[]", "default": 0, "context": true }, { "name": "dilation", "type": "int64[]", "default": 1, "context": true }, { "name": "groups", "type": "int64", "default": 1, "context": true }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv2d:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv2d:new", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1, "context": true }, { "name": "padding", "type": "int64[]", "default": 0, "context": true }, { "name": "dilation", "type": "int64[]", "default": 1, "context": true }, { "name": "groups", "type": "int64", "default": 1, "context": true }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv3d:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv3d:new", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1, "context": true }, { "name": "padding", "type": "int64[]", "default": 0, "context": true }, { "name": "dilation", "type": "int64[]", "default": 1, "context": true }, { "name": "groups", "type": "int64", "default": 1, "context": true }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv1d_relu:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1, "context": true }, { "name": "padding", "type": "int64[]", "default": 0, "context": true }, { "name": "dilation", "type": "int64[]", "default": 1, "context": true }, { "name": "groups", "type": "int64", "default": 1, "context": true }, { "name": "output_scale", "type": "float64" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv2d_relu:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "output_scale", "type": "float64" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv2d_relu:new", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1, "context": true }, { "name": "padding", "type": "int64[]", "default": 0, "context": true }, { "name": "dilation", "type": "int64[]", "default": 1, "context": true }, { "name": "groups", "type": "int64", "default": 1, "context": true }, { "name": "output_scale", "type": "float64" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv3d_relu:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1 }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "groups", "type": "int64", "default": 1 }, { "name": "output_scale", "type": "float64" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.conv3d_relu:new", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "context": true }, { "name": "bias", "type": "Tensor", "context": true }, { "name": "stride", "type": "int64[]", "default": 1, "context": true }, { "name": "padding", "type": "int64[]", "default": 0, "context": true }, { "name": "dilation", "type": "int64[]", "default": 1, "context": true }, { "name": "groups", "type": "int64", "default": 1, "context": true }, { "name": "output_scale", "type": "float64" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.add_relu:", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "qb", "type": "Tensor" }, { "name": "scale", "type": "float32" }, { "name": "zero_point", "type": "int64" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.add_relu:Scalar", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.cat", "category": "Tensor", "inputs": [ { "name": "qx", "type": "Tensor[]" }, { "name": "dim", "type": "int64" }, { "name": "scale", "type": "float32", "optional": true }, { "name": "zero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.cat_relu", "category": "Tensor", "inputs": [ { "name": "qx", "type": "Tensor[]" }, { "name": "dim", "type": "int64" }, { "name": "scale", "type": "float32", "optional": true }, { "name": "zero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm:", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm_relu", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm1d", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm1d_relu", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm2d", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm2d_relu", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm3d", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.batch_norm3d_relu", "category": "Normalization", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "mean", "type": "Tensor" }, { "name": "var", "type": "Tensor" }, { "name": "eps", "type": "float32" }, { "name": "output_scale", "type": "float32", "optional": true }, { "name": "outputzero_point", "type": "int64", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max_pool1d", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "ceil_mode", "type": "boolean", "default": false } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max_pool2d", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "ceil_mode", "type": "boolean", "default": false } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max_pool3d", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "ceil_mode", "type": "boolean", "default": false } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max_unpool2d", "category": "Pool", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "indices", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max_unpool3d", "category": "Pool", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "indices", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max_pool1d_with_indices", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "ceil_mode", "type": "boolean", "default": false }, { "name": "return_indices", "type": "boolean", "default": false } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.max_pool2d_with_indices", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "dilation", "type": "int64[]", "default": 1 }, { "name": "ceil_mode", "type": "boolean", "default": false }, { "name": "return_indices", "type": "boolean", "default": false } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.adaptive_avg_pool1d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.adaptive_avg_pool2d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.adaptive_avg_pool3d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.adaptive_max_pool1d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.adaptive_max_pool2d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.adaptive_max_pool3d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.atan", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.atan_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.atanh", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.atanh_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.arctan", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.arctan_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.avg_pool1d", "category": "Pool", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]", "default": 0 }, { "name": "ceil_mode", "type": "boolean", "default": false }, { "name": "count_include_pad", "type": "boolean", "default": true }, { "name": "divisor_override" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.avg_pool2d", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]" }, { "name": "ceil_mode", "type": "boolean", "default": false }, { "name": "count_include_pad", "type": "boolean", "default": true }, { "name": "divisor_override", "type": "int64", "optional": true } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.avg_pool3d", "category": "Pool", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "padding", "type": "int64[]" }, { "name": "ceil_mode", "type": "boolean", "default": false }, { "name": "count_include_pad", "type": "boolean", "default": true }, { "name": "divisor_override", "type": "int64", "optional": true } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.batch_norm", "category": "Normalization", "attributes": [ { "name": "training", "type": "boolean", "visible": false }, { "name": "momentum", "type": "float32", "default": 0.1 }, { "name": "eps", "type": "float32", "default": 0.00001 }, { "name": "cudnn_enabled", "type": "boolean", "visible": false } ], "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "running_mean", "type": "Tensor", "optional": true }, { "name": "running_var", "type": "Tensor", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.gather:", "inputs": [ { "name": "selm", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "sparse_grad", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.gather:dimname", "inputs": [ { "name": "selm", "type": "Tensor" }, { "name": "dim", "type": "Dimname" }, { "name": "index", "type": "Tensor" }, { "name": "sparse_grad", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.group_norm", "category": "Normalization", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "num_groups", "type": "int64" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "eps", "type": "float32", "default": 0.00001 }, { "name": "cudnn_enabled", "type": "boolean", "visible": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.relu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.relu_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.relu6", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.relu6_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.prelu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "weight", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardtanh", "category": "Activation", "attributes": [ { "name": "min_value", "type": "float64" }, { "name": "max_value", "type": "float64" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardtanh_", "category": "Activation", "attributes": [ { "name": "min_value", "type": "float64" }, { "name": "max_value", "type": "float64" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardswish", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardswish_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardsigmoid", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardsigmoid_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sigmoid", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sigmoid_", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.threshold", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "threshold", "type": "float64", "default": 0 }, { "name": "value", "type": "float64", "default": 0 }, { "name": "inplace", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.threshold_", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "threshold", "type": "float64" }, { "name": "value", "type": "float64" }, { "name": "inplace", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tile", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dims", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.softmax:int", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.softmax:Dimname", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "dim", "type": "Dimname" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log_softmax:int", "category": "Activation", "attributes": [ { "name": "dim", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log_softmax:Dimname", "category": "Activation", "attributes": [ { "name": "dim", "type": "Dimname" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tanh", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.take", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "index", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.leaky_relu", "category": "Activation", "attributes": [ { "name": "negative_slope", "type": "Scalar", "default": 0.01 } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.slice:Tensor", "category": "Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": 0 }, { "name": "start", "type": "int64", "optional": true, "default": null }, { "name": "end", "type": "int64", "optional": true, "default": null }, { "name": "step", "type": "int64", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.dropout", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64", "default": 0.5 }, { "name": "train", "type": "boolean", "visible": false } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.dropout_", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64", "default": 0.5 }, { "name": "train", "type": "boolean", "visible": false } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.feature_dropout", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64" }, { "name": "train", "type": "boolean" } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.feature_dropout_", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64" }, { "name": "train", "type": "boolean" } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.alpha_dropout", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64" }, { "name": "train", "type": "boolean" } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.alpha_dropout_", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64" }, { "name": "train", "type": "boolean" } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.feature_alpha_dropout", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64" }, { "name": "train", "type": "boolean" } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.feature_alpha_dropout_", "category": "Dropout", "attributes": [ { "name": "p", "type": "float64" }, { "name": "train", "type": "boolean" } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mm", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mat2", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.addmm", "category": "Layer", "attributes": [ { "name": "alpha", "type": "Scalar", "default": 1 }, { "name": "beta", "type": "Scalar", "default": 1 } ], "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mat1", "type": "Tensor" }, { "name": "mat2", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.add_:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.add_:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._add_relu:Tensor", "attributes": [ { "name": "alpha", "type": "Scalar", "default": 1 } ], "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._add_relu_:Tensor", "attributes": [ { "name": "alpha", "type": "Scalar", "default": 1 } ], "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sub_:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sub_:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pow:Tensor_Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "exponent", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pow:Scalar", "inputs": [ { "name": "self", "type": "Scalar" }, { "name": "exponent", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pow:Tensor_Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "exponent", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pow_:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "exponent", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pow_:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "exponent", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.add:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.add:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sub:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rsub:Scalar", "attributes": [ { "name": "other", "type": "Scalar" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sub:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rsub:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.diag:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "diagonal", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.block_diag:", "inputs": [ { "name": "tensors", "type": "Tensor[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.div:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.div:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.floordiv", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sum:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sum:dim_IntList", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "keepdim", "type": "boolean", "default": false }, { "name": "dtype", "type": "ScalarType", "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.matmul", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.transpose", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "dim0", "type": "int64" }, { "name": "dim1", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.chunk", "attributes": [ { "name": "chunks", "type": "int64" }, { "name": "dim", "type": "int64", "default": 0 } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.unsafe_chunk", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "chunks", "type": "int64" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.view:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.view:SymInt", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "SymInt[]", "visible": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.to:dtype_layout", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "non_blocking", "type": "boolean", "default": false }, { "name": "copy", "type": "boolean", "default": false }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.to:device", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "device", "type": "Device" }, { "name": "dtype", "type": "ScalarType" }, { "name": "non_blocking", "type": "boolean", "default": false }, { "name": "copy", "type": "boolean", "default": false }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.to:dtype", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType" }, { "name": "non_blocking", "type": "boolean", "default": false }, { "name": "copy", "type": "boolean", "default": false }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.to:other", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false }, { "name": "copy", "type": "boolean", "default": false }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clip:", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "min", "type": "Scalar", "optional": true, "default": null }, { "name": "max", "type": "Scalar", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clip:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "min", "type": "Tensor", "optional": true, "default": null }, { "name": "max", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clip_:", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "min", "type": "Scalar", "optional": true, "default": null }, { "name": "max", "type": "Scalar", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clip_:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "min", "type": "Tensor", "optional": true, "default": null }, { "name": "max", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clone", "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.complex", "inputs": [ { "name": "real", "type": "Tensor" }, { "name": "imag", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.expand", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "implicit", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.copy_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "src", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.reshape", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "shape", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rot90", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "k", "type": "int64", "default": 1 }, { "name": "dims", "type": "int64[]", "default": [ 0, 1 ] } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.round:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.round_:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.round:decimals", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "decimals", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.round_:decimals", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "decimals", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.select:int", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.select:Dimname", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" }, { "name": "index", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.repeat", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "repeats", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mean:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mean:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "keepdim", "type": "boolean", "default": false }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.median:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.median:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "values", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.moveaxis:int", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "source", "type": "int64" }, { "name": "destination", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.moveaxis:intlist", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "source", "type": "int64[]" }, { "name": "destination", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.lstm:input", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "hx", "type": "Tensor[]" }, { "name": "params", "type": "Tensor[]" }, { "name": "has_biases", "type": "boolean" }, { "name": "num_layers", "type": "int64" }, { "name": "dropout", "type": "float64" }, { "name": "train", "type": "boolean" }, { "name": "bidirectional", "type": "boolean" }, { "name": "batch_first", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.lstm:data", "category": "Layer", "inputs": [ { "name": "data", "type": "Tensor" }, { "name": "batch_sizes", "type": "Tensor" }, { "name": "hx", "type": "Tensor[]" }, { "name": "params", "type": "Tensor[]" }, { "name": "has_biases", "type": "boolean" }, { "name": "num_layers", "type": "int64" }, { "name": "dropout", "type": "float64" }, { "name": "train", "type": "boolean" }, { "name": "bidirectional", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.gru:input", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "hx", "type": "Tensor" }, { "name": "params", "type": "Tensor[]" }, { "name": "has_biases", "type": "boolean" }, { "name": "num_layers", "type": "int64" }, { "name": "dropout", "type": "float64" }, { "name": "train", "type": "boolean" }, { "name": "bidirectional", "type": "boolean" }, { "name": "first_batch", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.gru:data", "category": "Layer", "inputs": [ { "name": "data", "type": "Tensor" }, { "name": "batch_sizes", "type": "Tensor" }, { "name": "hx", "type": "Tensor" }, { "name": "params", "type": "Tensor[]" }, { "name": "has_biases", "type": "boolean" }, { "name": "num_layers", "type": "int64" }, { "name": "dropout", "type": "float64" }, { "name": "train", "type": "boolean" }, { "name": "bidirectional", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.coalesce", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._coalesce", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.contiguous", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "memory_format", "type": "MemoryFormat", "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.squeeze:", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.squeeze:dim", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.squeeze:dimname", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.squeeze_:", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.squeeze_:dim", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.squeeze_:dimname", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unsqueeze", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unsqueeze_", "category": "Transform", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.max:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": 0 }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "values", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.max:other", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.maximum", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.min:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.min:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": 0 }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "values", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.min:other", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.minimum", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mish_:", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mish:", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cat:", "category": "Tensor", "inputs": [ { "name": "tensors", "type": "Tensor[]" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cat:names", "category": "Tensor", "inputs": [ { "name": "tensors", "type": "Tensor[]" }, { "name": "dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.concat:", "category": "Tensor", "inputs": [ { "name": "tensors", "type": "Tensor[]" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.concat:names", "category": "Tensor", "inputs": [ { "name": "tensors", "type": "Tensor[]" }, { "name": "dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.broadcast_tensors", "inputs": [ { "name": "tensors", "type": "Tensor[]" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.broadcast_to:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.meshgrid:", "category": "Tensor", "inputs": [ { "name": "tensors", "type": "Tensor[]" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.meshgrid:indexing", "category": "Tensor", "inputs": [ { "name": "tensors", "type": "Tensor[]" }, { "name": "indexing", "type": "string" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.stack:", "category": "Tensor", "inputs": [ { "name": "inputs", "type": "Tensor[]" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.expand_as", "inputs": [ { "name": "inputs", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.eye:", "inputs": [ { "name": "n", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.eye:m", "inputs": [ { "name": "n", "type": "int64" }, { "name": "m", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pairwise_distance:", "inputs": [ { "name": "x1", "type": "Tensor" }, { "name": "x2", "type": "Tensor" }, { "name": "p", "type": "float32", "default": 2 }, { "name": "eps", "type": "float32", "default": 1e-06 }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pad:", "category": "Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "pad", "type": "int64[]" }, { "name": "mode", "type": "string", "default": "constant" }, { "name": "value", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.permute", "category": "Shape", "inputs": [ { "name": "inputs", "type": "Tensor" }, { "name": "dims", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.embedding", "category": "Transform", "inputs": [ { "name": "inputs", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "padding_idx", "type": "int64" }, { "name": "scale_grad_by_freq", "type": "boolean", "default": false }, { "name": "sparse", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.embedding_bag", "category": "Transform", "inputs": [ { "name": "weight", "type": "Tensor" }, { "name": "indices", "type": "Tensor" }, { "name": "offsets", "type": "Tensor" }, { "name": "scale_grad_by_freq", "type": "boolean", "default": false }, { "name": "mode", "type": "int64", "default": 0 }, { "name": "sparse", "type": "boolean", "default": false }, { "name": "per_sample_weights", "type": "Tensor", "optional": true, "default": null }, { "name": "include_last_offset", "type": "boolean", "default": false } ], "outputs": [ { "name": "output1", "type": "Tensor" }, { "name": "output2", "type": "Tensor" }, { "name": "output3", "type": "Tensor" }, { "name": "output4", "type": "Tensor" } ] }, { "name": "torch.flatten:using_ints", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "start_dim", "type": "int64", "default": 0 }, { "name": "end_dim", "type": "int64", "default": -1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.flatten:named_out_dim", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "start_dim", "type": "int64" }, { "name": "end_dim", "type": "int64" }, { "name": "out_dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.flatten:using_names", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "start_dim", "type": "Dimname" }, { "name": "end_dim", "type": "Dimname" }, { "name": "out_dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.flatten:DimnameList", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dims", "type": "Dimname[]" }, { "name": "out_dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unflatten:int", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "sizes", "type": "int64[]" }, { "name": "names", "type": "Dimname[]", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unflatten:Dimname", "category": "Shape", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" }, { "name": "sizes", "type": "int64[]" }, { "name": "names", "type": "Dimname[]", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._pack_padded_sequence", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "lengths", "type": "Tensor" }, { "name": "batch_first", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch._pad_packed_sequence", "inputs": [ { "name": "data", "type": "Tensor" }, { "name": "batch_sizes", "type": "Tensor" }, { "name": "batch_first", "type": "boolean" }, { "name": "padding_value", "type": "Scalar" }, { "name": "total_length", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.quantize_per_tensor:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "scale", "type": "float32" }, { "name": "zero_point", "type": "int64" }, { "name": "dtype", "type": "ScalarType" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.quantize_per_tensor:qparams", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "scale", "type": "Tensor" }, { "name": "zero_point", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.quantize_per_tensor:tensors", "inputs": [ { "name": "tensors", "type": "Tensor[]" }, { "name": "scales", "type": "Tensor" }, { "name": "zero_points", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType" } ], "outputs": [ { "name": "outputs", "type": "Tensor[]" } ] }, { "name": "torch.floor_divide", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.index", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "indices", "type": "Tensor[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Byte", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Char", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Short", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Int", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Long", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Half", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Float", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._cast_Double", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "non_blocking", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.frobenius_norm:", "category": "Normalization", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.frobenius_norm:dim", "category": "Normalization", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.norm:ScalarOpt_dtype", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "p", "type": "Scalar", "optional": true }, { "name": "dtype", "type": "ScalarType" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.norm:ScalarOpt_dim_dtype", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "p", "type": "Scalar", "optional": true }, { "name": "dim", "type": "int64[1]" }, { "name": "keepdim", "type": "boolean" }, { "name": "dtype", "type": "ScalarType" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.norm:ScalarOpt_dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "p", "type": "Scalar", "optional": true }, { "name": "dim", "type": "int64[1]" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.norm:dtype_out", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "p", "type": "Scalar", "optional": true }, { "name": "dim", "type": "int64[1]" }, { "name": "keepdim", "type": "boolean", "default": false }, { "name": "dtype", "type": "ScalarType" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.norm:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "p", "type": "Scalar", "default": 2 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.numpy_T", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.glu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": -1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.gelu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "approximate", "type": "string", "default": "none" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.gelu_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "approximate", "type": "string", "default": "none" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.celu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.elu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 }, { "name": "scale", "type": "Scalar", "default": 1 }, { "name": "input_scale", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.elu_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "alpha", "type": "Scalar", "default": 1 }, { "name": "scale", "type": "Scalar", "default": 1 }, { "name": "input_scale", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.selu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.selu_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.silu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.silu_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardswish", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hardswish_", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pixel_shuffle", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "upscale_factor", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.nonzero", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.neg", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ne:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.bitwise_not", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.bitwise_not_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.bucketize:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "boundaries", "type": "Tensor" }, { "name": "out_int32", "type": "boolean", "default": false }, { "name": "right", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.bucketize:Scalar", "inputs": [ { "name": "self", "type": "Scalar" }, { "name": "boundaries", "type": "Tensor" }, { "name": "out_int32", "type": "boolean", "default": false }, { "name": "right", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pinverse", "inputs": [ { "name": "self", "type": "Scalar" }, { "name": "rcond", "type": "float32", "default": 1e-15 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.real", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.imag", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.conj", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._conj", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rsqrt", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.grid_sampler", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "grid", "type": "Tensor" }, { "name": "interpolation_mode", "type": "int64" }, { "name": "padding_mode", "type": "int64" }, { "name": "align_corners", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sort", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": -1 }, { "name": "descending", "type": "boolean", "default": false } ], "outputs": [ { "name": "values", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.gt:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.gt:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mul:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mul:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mul:ScalarT", "inputs": [ { "name": "input", "type": "Tensor[]" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.mul_:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mul_:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.multinomial", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "num_samples", "type": "int64" }, { "name": "replacement", "type": "boolean", "default": false }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.div_:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.div_:Scalar", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.bmm", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "mat2", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.constant_pad_nd", "category": "Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "pad", "type": "int64[]" }, { "name": "value", "type": "Scalar", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ones", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.empty:names", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "names", "type": "Dimname[]", "optional": true, "default": null }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.empty:memory_format", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.zeros", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.zeros_like", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ones_like", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.amax:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]", "default": [] }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.amin:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]", "default": [] }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._aminmax:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "min", "type": "Tensor" }, { "name": "max", "type": "Tensor" } ] }, { "name": "torch._aminmax:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "min", "type": "Tensor" }, { "name": "max", "type": "Tensor" } ] }, { "name": "torch.aminmax:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "optional": true, "default": null }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "min", "type": "Tensor" }, { "name": "max", "type": "Tensor" } ] }, { "name": "torch.arange:", "inputs": [ { "name": "end", "type": "Scalar" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.arange:start", "inputs": [ { "name": "start", "type": "Scalar" }, { "name": "end", "type": "Scalar" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.arange:start_step", "inputs": [ { "name": "start", "type": "Scalar" }, { "name": "end", "type": "Scalar" }, { "name": "step", "type": "Scalar" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.arange:start_out", "inputs": [ { "name": "start", "type": "Scalar" }, { "name": "end", "type": "Scalar" }, { "name": "step", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clamp", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "min", "type": "Scalar", "optional": true, "default": null }, { "name": "max", "type": "Scalar", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clamp_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "min", "type": "Scalar", "optional": true, "default": null }, { "name": "max", "type": "Scalar", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clamp_min", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "min", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clamp_min_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "min", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clamp_max", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "max", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clamp_max_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "max", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fill_:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "value", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fill_:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "value", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.floor", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.floor_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.floor_divide", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.floor_divide_:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.floor_divide_:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.abs", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.abs_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.leaky_relu_", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "negative_slope", "type": "Scalar", "default": 0.01 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.instance_norm", "category": "Normalization", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true }, { "name": "bias", "type": "Tensor", "optional": true }, { "name": "running_mean", "type": "Tensor", "optional": true }, { "name": "running_var", "type": "Tensor", "optional": true }, { "name": "use_input_stats", "type": "boolean" }, { "name": "momentum", "type": "float32" }, { "name": "eps", "type": "float32" }, { "name": "cudnn_enabled", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.reflection_pad1d", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.replication_pad1d", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.replication_pad2d", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.replication_pad3d", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.reflection_pad2d", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.reflection_pad3d", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "padding", "type": "int64[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rnn_relu:input", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "hx", "type": "Tensor" }, { "name": "params", "type": "Tensor[]" }, { "name": "has_biases", "type": "boolean" }, { "name": "num_layers", "type": "int64" }, { "name": "dropout", "type": "float32" }, { "name": "train", "type": "boolean" }, { "name": "bidirectional", "type": "boolean" }, { "name": "batch_first", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.rnn_tanh:input", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "hx", "type": "Tensor" }, { "name": "params", "type": "Tensor[]" }, { "name": "has_biases", "type": "boolean" }, { "name": "num_layers", "type": "int64" }, { "name": "dropout", "type": "float32" }, { "name": "train", "type": "boolean" }, { "name": "bidirectional", "type": "boolean" }, { "name": "batch_first", "type": "boolean" } ], "outputs": [ { "name": "output" }, { "name": "?" } ] }, { "name": "torch.roll:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "shifts", "type": "int64[]" }, { "name": "dims", "type": "int64[]", "default": [] } ], "outputs": [ { "name": "output" } ] }, { "name": "torch.detach", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unique_consecutive:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "return_inverse", "type": "boolean", "default": false }, { "name": "return_counts", "type": "boolean", "default": false }, { "name": "dim", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output1", "type": "Tensor" }, { "name": "output2", "type": "Tensor" }, { "name": "output3", "type": "Tensor" } ] }, { "name": "torch.unique_dim_consecutive:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "return_inverse", "type": "boolean", "default": false }, { "name": "return_counts", "type": "boolean", "default": false } ], "outputs": [ { "name": "output1", "type": "Tensor" }, { "name": "output2", "type": "Tensor" }, { "name": "output3", "type": "Tensor" } ] }, { "name": "torch.upsample_bicubic2d:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "align_corners", "type": "boolean" }, { "name": "scales_h", "type": "float32", "optional": true, "default": null }, { "name": "scales_w", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_bicubic2d:vec", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "align_corners", "type": "boolean" }, { "name": "scales_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_linear1d:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "align_corners", "type": "boolean" }, { "name": "scales", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_linear1d:vec", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": true }, { "name": "align_corners", "type": "boolean" }, { "name": "scale_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_bilinear2d:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "align_corners", "type": "boolean" }, { "name": "scales_h", "type": "float32", "optional": true, "default": null }, { "name": "scales_w", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_bilinear2d:vec", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": "true" }, { "name": "align_corners", "type": "boolean" }, { "name": "scales_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_bilinear2d:vec", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": true }, { "name": "align_corners", "type": "boolean" }, { "name": "scale_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_nearest1d:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "scales", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_nearest1d:vec", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": true }, { "name": "scales_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_nearest2d:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "scales_h", "type": "float32", "optional": true, "default": null }, { "name": "scales_w", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_nearest2d:vec", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": true }, { "name": "scales_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_nearest3d:", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "scales_d", "type": "float32", "optional": true, "default": null }, { "name": "scales_h", "type": "float32", "optional": true, "default": null }, { "name": "scales_w", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_nearest3d:vec", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": true }, { "name": "scales_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_trilinear3d:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_size", "type": "int64[]" }, { "name": "align_corners", "type": "boolean" }, { "name": "scales_d", "type": "float32", "optional": true, "default": null }, { "name": "scales_h", "type": "float32", "optional": true, "default": null }, { "name": "scales_w", "type": "float32", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.upsample_trilinear3d:vec", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_size", "type": "int64[]", "optional": true }, { "name": "align_corners", "type": "boolean" }, { "name": "scale_factors", "type": "float32[]", "optional": true } ], "outputs": [ { "name": "output", "type": "int64" } ] }, { "name": "torch.size", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "output", "type": "int64" } ] }, { "name": "torch.exp", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.exp_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.reciprocal", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.square", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.square_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sqrt", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.std:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "unbiased", "type": "boolean", "default": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.std:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "unbiased", "type": "boolean", "default": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.std_mean:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "unbiased", "type": "boolean", "default": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.std_mean:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "unbiased", "type": "boolean", "default": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.erf", "inputs": [ { "name": "input", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.argmin", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.argmax", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log10", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log1p", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log2", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.logdet", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.affine_grid_generator", "inputs": [ { "name": "theta", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "align_corners", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__interpolate", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "size", "type": "int64" }, { "name": "scale_factor", "type": "float32[]" }, { "name": "mode", "type": "string" }, { "name": "align_corners", "type": "boolean" }, { "name": "recompute_scale_factor", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rand:", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rand:generator", "inputs": [ { "name": "high", "type": "int64", "optional": true, "default": null }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.rand_link", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "generator", "type": "Generator", "optional": true, "default": null }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.randint:generator", "inputs": [ { "name": "high", "type": "int64", "optional": true, "default": null }, { "name": "size", "type": "int64[]" }, { "name": "generator", "type": "Generator" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.randint:low", "inputs": [ { "name": "low", "type": "int64", "optional": true, "default": null }, { "name": "high", "type": "int64", "optional": true, "default": null }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.randn", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unsafe_split:Tensor", "category": "Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "split_size", "type": "int64" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "outputs", "type": "Tensor[]" } ] }, { "name": "torch._sparse_coo_tensor_unsafe:", "category": "Tensor", "inputs": [ { "name": "indices", "type": "Tensor" }, { "name": "values", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "outputs", "type": "Tensor[]" } ] }, { "name": "torch.split:Tensor", "category": "Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "split_size", "type": "int64" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "outputs", "type": "Tensor[]" } ] }, { "name": "torch.split:Tensor_sizes", "category": "Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "split_sizes", "type": "int64[]" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "outputs", "type": "Tensor[]" } ] }, { "name": "torch.split_with_sizes", "category": "Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "split_sizes", "type": "int64[]" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "outputs", "type": "Tensor[]" } ] }, { "name": "torch.remainder:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.remainder:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.remainder_:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.remainder_:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.layer_norm", "category": "Normalization", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "normalized_shape", "type": "int64[]" }, { "name": "weight", "type": "Tensor", "optional": true, "default": null }, { "name": "bias", "type": "Tensor", "optional": true, "default": null }, { "name": "eps", "type": "float32", "default": 0.00001 }, { "name": "cudnn_enable", "type": "boolean", "default": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter:src", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter_:src", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter:value", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter_:value", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter:reduce", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Tensor" }, { "name": "reduce", "type": "string" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter_:reduce", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Tensor" }, { "name": "reduce", "type": "string" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter:value_reduce", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Scalar" }, { "name": "reduce", "type": "string" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter_:value_reduce", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Scalar" }, { "name": "reduce", "type": "string" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.scatter_add_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64"}, { "name": "index", "type": "Tensor" }, { "name": "src", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sin", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sinh", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cos", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cosh", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ctc_loss:IntList", "inputs": [ { "name": "log_probs", "type": "Tensor" }, { "name": "targets", "type": "Tensor" }, { "name": "input_lengths", "type": "int64[]" }, { "name": "target_lengths", "type": "int64[]" }, { "name": "blank", "type": "int64", "default": 0 }, { "name": "reduction", "type": "int64" }, { "name": "zero_infinity", "type": "boolean", "default": "false" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._ctc_loss:", "inputs": [ { "name": "log_probs", "type": "Tensor" }, { "name": "targets", "type": "Tensor" }, { "name": "input_lengths", "type": "int64[]" }, { "name": "target_lengths", "type": "int64[]" }, { "name": "blank", "type": "int64", "default": 0 }, { "name": "zero_infinity", "type": "boolean", "default": "false" } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tan", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.atan2:Tensor", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tan_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tanh", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tanh_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cdist:", "inputs": [ { "name": "x1", "type": "Tensor" }, { "name": "x2", "type": "Tensor" }, { "name": "p", "type": "float32", "default": 2 }, { "name": "compute_mode", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.pdist:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "p", "type": "float32", "default": 2 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.narrow:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "start", "type": "int64" }, { "name": "length", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.narrow:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "start", "type": "Tensor" }, { "name": "length", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.narrow_copy", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "start", "type": "int64" }, { "name": "length", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.index_select:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "index", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.index_put", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "indices", "type": "Tensor[]", "optional": true }, { "name": "values", "type": "Tensor" }, { "name": "accumulate", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.index_put_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "indices", "type": "Tensor[]", "optional": true }, { "name": "values", "type": "Tensor" }, { "name": "accumulate", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.new_zeros", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.new_full", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "fill_value", "type": "Scalar" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.softplus", "category": "Activation", "attributes": [ { "name": "beta", "type": "Scalar", "default": 1 }, { "name": "threshold", "type": "Scalar", "default": 20 } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.zero_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cumsum:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cumsum:dimname", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cumsum_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.masked_fill:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mask", "type": "Tensor" }, { "name": "value", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.masked_fill_:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mask", "type": "Tensor" }, { "name": "value", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.masked_fill:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mask", "type": "Tensor" }, { "name": "value", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.masked_fill_:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mask", "type": "Tensor" }, { "name": "value", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.masked_select:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mask", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.argmax", "attributes": [ { "name": "dim", "type": "int64", "optional": true, "default": null }, { "name": "keepdim", "type": "boolean", "default": false } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.argmin", "attributes": [ { "name": "dim", "type": "int64", "optional": true, "default": null }, { "name": "keepdim", "type": "boolean", "default": false } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "signal_ndim", "type": "int64" }, { "name": "normalized", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_fft2", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_ifft2", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_rfft2", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_irfft2", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_hfft2", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_ihfft2", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_fftn", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_ifftn", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_rfftn", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_irfftn", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_hfftn", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fft_ihfftn", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "s", "s": "int64[]", "optional": true, "default": null }, { "name": "dim", "dim": "int64[]", "optional": true, "default": [ -2, -1] }, { "name": "norm", "type": "string", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.to_mkldnn:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.topk:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "k", "type": "int64" }, { "name": "dim", "type": "int64", "default": -1 }, { "name": "largest", "type": "boolean", "default": true }, { "name": "sorted", "type": "boolean", "default": true } ], "outputs": [ { "name": "values", "type": "Tensor" }, { "name": "indices", "type": "Tensor" } ] }, { "name": "torch.random_:", "attributes": [ { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.random_:to", "attributes": [ { "name": "to", "type": "int64" }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.random_:from", "attributes": [ { "name": "from", "type": "int64" }, { "name": "to", "type": "int64", "optional": true }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.uniform_", "attributes": [ { "name": "from", "type": "float32", "default": 0 }, { "name": "to", "type": "float32", "default": 1 }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cauchy_", "attributes": [ { "name": "median", "type": "float32", "default": 0 }, { "name": "sigma", "type": "float32", "default": 1 }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.log_normal_", "attributes": [ { "name": "mean", "type": "float32", "default": 1 }, { "name": "std", "type": "float32", "default": 2 }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.exponential_", "attributes": [ { "name": "lambd", "type": "float32", "default": 1 }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.geometric_", "attributes": [ { "name": "p", "type": "float32" }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mv", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "vec", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mvlgamma", "attributes": [ { "name": "p", "type": "int64" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.mvlgamma_", "attributes": [ { "name": "p", "type": "int64" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.quantize_per_channel", "attributes": [ { "name": "axis", "type": "int64" }, { "name": "dtype", "type": "ScalarType" } ], "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "scales", "type": "Tensor" }, { "name": "zero_points", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._shape_as_tensor", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.clone", "attributes": [ { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.im2col", "attributes": [ { "name": "kernel_size", "type": "int64[]" }, { "name": "dilation", "type": "int64[]" }, { "name": "padding", "type": "int64[]" }, { "name": "stride", "type": "int64[]" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.inverse:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.col2im", "attributes": [ { "name": "output_size", "type": "int64[]" }, { "name": "kernel_size", "type": "int64[]" }, { "name": "dilation", "type": "int64[]" }, { "name": "padding", "type": "int64[]" }, { "name": "stride", "type": "int64[]" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.eq:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.eq:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.where:self", "inputs": [ { "name": "condition", "type": "Tensor" }, { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__and__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__or__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ge:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ge:Scalar", "attributes": [ { "name": "other", "type": "Scalar" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.lt:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.lt:Scalar", "attributes": [ { "name": "other", "type": "Scalar" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.flip", "attributes": [ { "name": "dims", "type": "int64[]" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.linalg_solve:", "attributes": [ { "name": "input", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.linalg_tensorinv:", "attributes": [ { "name": "self", "type": "Tensor" }, { "name": "ind", "type": "int64", "default": 2 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.linalg_tensorsolve:", "attributes": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "dims", "type": "int64[]", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.linspace", "attributes": [ { "name": "start", "type": "Scalar" }, { "name": "end", "type": "Scalar" }, { "name": "steps", "type": "int64", "optional": true, "default": null }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sign", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.sign_", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.signbit", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.dist", "attributes": [ { "name": "start", "type": "Scalar", "default": 2 } ], "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ceil:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.ceil_:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.einsum", "inputs": [ { "name": "equation", "type": "string" }, { "name": "tensors", "type": "Tensor[]" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.dot", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "tensor", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.vdot", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.var:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "unbiased", "type": "boolean", "default": true } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.var:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "unbiased", "type": "boolean", "default": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.var:correction", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]", "optional": true }, { "name": "correction", "type": "int64", "optional": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.var:names_dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname[]" }, { "name": "unbiased", "type": "boolean", "default": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.var:correction_names", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname[]" }, { "name": "correction", "type": "int64", "default": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.var_mean:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "unbiased", "type": "boolean", "default": true } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.var_mean:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64[]" }, { "name": "unbiased", "type": "boolean", "default": true }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "?", "type": "Tensor" } ] }, { "name": "torch.all:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.all:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.any:", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.any:dim", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64" }, { "name": "keepdim", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.renorm", "attributes": [ { "name": "p", "type": "Scalar" }, { "name": "dim", "type": "int64" }, { "name": "maxnorm", "type": "Scalar" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unfold", "attributes": [ { "name": "dimension", "type": "int64" }, { "name": "size", "type": "int64" }, { "name": "step", "type": "int64" } ], "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fake_quantize_per_channel_affine", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "scale", "type": "Tensor" }, { "name": "zero_point", "type": "Tensor" }, { "name": "axis", "type": "int64" }, { "name": "quant_min", "type": "int64" }, { "name": "quant_max", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.fake_quantize_per_tensor_affine", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "scale", "type": "float32" }, { "name": "zero_point", "type": "int64" }, { "name": "quant_min", "type": "int64" }, { "name": "quant_max", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.addmv", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mat", "type": "Tensor" }, { "name": "vec", "type": "Tensor" }, { "name": "beta", "type": "Scalar", "default": 1 }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.addmv_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mat", "type": "Tensor" }, { "name": "vec", "type": "Tensor" }, { "name": "beta", "type": "Scalar", "default": 1 }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.type_as", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.normal_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mean", "type": "float32", "default": 0 }, { "name": "std", "type": "float32", "default": 1 }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.normal:Tensor_float", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "mean", "type": "float32", "default": 0 }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.normal:float_Tensor", "inputs": [ { "name": "mean", "type": "float32" }, { "name": "std", "type": "Tensor" }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.normal:Tensor_Tensor", "inputs": [ { "name": "mean", "type": "Tensor" }, { "name": "std", "type": "Tensor" }, { "name": "generator", "type": "Generator", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.normal:float_float", "inputs": [ { "name": "mean", "type": "float32", "default": 0 }, { "name": "std", "type": "float32" }, { "name": "size", "type": "int64[]" }, { "name": "generator", "type": "Generator", "optional": true, "default": null }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.empty_like", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.classes._nnapi.Compilation", "inputs": [ { "name": "serialized_model", "type": "Tensor" }, { "name": "inputs", "type": "Tensor[]" }, { "name": "parameter_buffers", "type": "Tensor[]" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.argsort:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": -1 }, { "name": "descending", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.argsort:dimname", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" }, { "name": "descending", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops._caffe2.CopyGPUToCPU", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "?" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops._caffe2.CopyCPUToGPU", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "?" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops._caffe2.GenerateProposals", "inputs": [ { "name": "scores", "type": "Tensor" }, { "name": "bbox_deltas", "type": "Tensor" }, { "name": "im_info", "type": "Tensor" }, { "name": "anchors", "type": "Tensor" }, { "name": "spatial_scale", "type": "float32" }, { "name": "pre_nms_topN", "type": "int64" }, { "name": "post_nms_topN", "type": "int64" }, { "name": "nms_thresh", "type": "float32" }, { "name": "min_size", "type": "float32" }, { "name": "angle_bound_on", "type": "boolean" }, { "name": "angle_bound_lo", "type": "int64" }, { "name": "angle_bound_hi", "type": "int64" }, { "name": "clip_angle_thresh", "type": "float32" }, { "name": "legacy_plus_one", "type": "boolean" }, { "name": "?" } ], "outputs": [ { "name": "output_0", "type": "Tensor" }, { "name": "output_1", "type": "Tensor" } ] }, { "name": "ops._caffe2.CollectRpnProposals", "inputs": [ { "name": "input_list", "type": "Tensor[]" }, { "name": "rpn_max_level", "type": "int64" }, { "name": "rpn_min_level", "type": "int64" }, { "name": "rpn_post_nms_topN", "type": "int64" }, { "name": "?" } ], "outputs": [ { "name": "rois", "type": "Tensor" } ] }, { "name": "ops._caffe2.DistributeFpnProposals", "inputs": [ { "name": "rois", "type": "Tensor" }, { "name": "roi_canonical_scale", "type": "int64" }, { "name": "roi_canonical_level", "type": "int64" }, { "name": "roi_max_level", "type": "int64" }, { "name": "roi_min_level", "type": "int64" }, { "name": "legacy_plus_one", "type": "boolean" }, { "name": "?" } ], "outputs": [ { "name": "rois_fpn2", "type": "Tensor" }, { "name": "rois_fpn3", "type": "Tensor" }, { "name": "rois_fpn4", "type": "Tensor" }, { "name": "rois_fpn5", "type": "Tensor" }, { "name": "rois_idx_restore_int32", "type": "Tensor" } ] }, { "name": "ops._caffe2.RoIAlign", "inputs": [ { "name": "features", "type": "Tensor" }, { "name": "rois", "type": "Tensor" }, { "name": "order", "type": "string" }, { "name": "spatial_scale", "type": "float32" }, { "name": "pooled_h", "type": "int64" }, { "name": "pooled_w", "type": "int64" }, { "name": "sampling_ratio", "type": "int64" }, { "name": "aligned", "type": "boolean" }, { "name": "?" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops._caffe2.BatchPermutation", "inputs": [ { "name": "X", "type": "Tensor" }, { "name": "indices", "type": "Tensor" }, { "name": "?" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops._caffe2.BBoxTransform", "inputs": [ { "name": "rois", "type": "Tensor" }, { "name": "deltas", "type": "Tensor" }, { "name": "im_info", "type": "Tensor" }, { "name": "weights", "type": "float32[]" }, { "name": "apply_scale", "type": "boolean" }, { "name": "rotated", "type": "boolean" }, { "name": "angle_bound_on", "type": "boolean" }, { "name": "angle_bound_lo", "type": "int64" }, { "name": "angle_bound_hi", "type": "int64" }, { "name": "clip_angle_thresh", "type": "float32" }, { "name": "legacy_plus_one", "type": "boolean" }, { "name": "?" } ], "outputs": [ { "name": "output_0", "type": "Tensor" }, { "name": "output_1", "type": "Tensor" } ] }, { "name": "ops._caffe2.BoxWithNMSLimit", "inputs": [ { "name": "scores", "type": "Tensor" }, { "name": "boxes", "type": "Tensor" }, { "name": "batch_splits", "type": "Tensor" }, { "name": "score_thresh", "type": "float32" }, { "name": "nms", "type": "float32" }, { "name": "detections_per_im", "type": "int64" }, { "name": "soft_nms_enabled", "type": "boolean" }, { "name": "soft_nms_method", "type": "string" }, { "name": "soft_nms_sigma", "type": "float32" }, { "name": "soft_nms_min_score_thres", "type": "float32" }, { "name": "rotated", "type": "boolean" }, { "name": "cls_agnostic_bbox_reg", "type": "boolean" }, { "name": "input_boxes_include_bg_cls", "type": "boolean" }, { "name": "output_classes_include_bg_cls", "type": "boolean" }, { "name": "legacy_plus_one", "type": "boolean" }, { "name": "?" } ], "outputs": [ { "name": "scores", "type": "Tensor" }, { "name": "boxes", "type": "Tensor" }, { "name": "classes", "type": "Tensor" }, { "name": "batch_splits", "type": "Tensor" }, { "name": "keeps", "type": "Tensor" }, { "name": "keeps_size", "type": "Tensor" } ] }, { "name": "ops.neuron.forward_v2_1", "inputs": [ { "name": "input", "type": "Tensor[]" }, { "name": "model" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.repeat_interleave:Tensor", "inputs": [ { "name": "repeats", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.repeat_interleave:self_Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "repeats", "type": "Tensor" }, { "name": "dim", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.repeat_interleave:self_int", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "repeats", "type": "int64" }, { "name": "dim", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cpu", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.hardswish", "category": "Activation", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.celu", "category": "Activation", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" }, { "name": "alpha", "type": "Scalar", "default": 1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.mul:", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "qb", "type": "Tensor" }, { "name": "scale", "type": "float32" }, { "name": "zero_point", "type": "int64" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.mul:Scalar", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.mul:Scalar2", "inputs": [ { "name": "b", "type": "Scalar" }, { "name": "qa", "type": "Tensor" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.mul_relu:Scalar", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "b", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.mul_relu:Scalar2", "inputs": [ { "name": "b", "type": "Scalar" }, { "name": "qa", "type": "Tensor" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.mul_scalar", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "qb", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.mul_scalar_relu", "inputs": [ { "name": "qa", "type": "Tensor" }, { "name": "qb", "type": "Scalar" } ], "outputs": [ { "name": "qc", "type": "Tensor" } ] }, { "name": "ops.quantized.relu6", "category": "Activation", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "inplace", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.leaky_relu", "category": "Activation", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "negative_slope", "type": "Scalar", "default": false }, { "name": "inplace", "type": "boolean" }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.quantized.sigmoid", "category": "Activation", "inputs": [ { "name": "qx", "type": "Tensor" }, { "name": "output_scale", "type": "float32" }, { "name": "output_zero_point", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.full:names", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "fill_value", "type": "Scalar" }, { "name": "names", "type": "Dimname[]", "optional": true }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.full:", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "fill_value", "type": "Scalar" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.full:out", "inputs": [ { "name": "size", "type": "int64[]" }, { "name": "fill_value", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.full_like:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "fill_value", "type": "Scalar" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null }, { "name": "memory_format", "type": "MemoryFormat", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.new_empty:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.new_empty_strided:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.linear:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "weight", "type": "Tensor" }, { "name": "bias", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._thnn_fused_gru_cell", "inputs": [ { "name": "input_gates", "type": "Tensor" }, { "name": "hidden_gates", "type": "Tensor" }, { "name": "hx", "type": "Tensor" }, { "name": "input_bias", "type": "Tensor", "optional": true, "default": null }, { "name": "hidden_bias", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "output2", "type": "Tensor" } ] }, { "name": "torch._thnn_fused_lstm_cell", "inputs": [ { "name": "input_gates", "type": "Tensor" }, { "name": "hidden_gates", "type": "Tensor" }, { "name": "cx", "type": "Tensor" }, { "name": "input_bias", "type": "Tensor", "optional": true, "default": null }, { "name": "hidden_bias", "type": "Tensor", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "output2", "type": "Tensor" }, { "name": "output3", "type": "Tensor" } ] }, { "name": "torch._unique2:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "sorted", "type": "boolean", "default": false }, { "name": "return_inverse", "type": "boolean", "default": false }, { "name": "return_counts", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" }, { "name": "output", "type": "Tensor" }, { "name": "output", "type": "Tensor" } ] }, { "name": "torch._weight_norm", "inputs": [ { "name": "v", "type": "Tensor" }, { "name": "g", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.requires_grad_", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "requires_grad", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.stft", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "n_fft", "type": "int64" }, { "name": "hop_length", "type": "int64", "optional": true, "default": null }, { "name": "win_length", "type": "int64", "optional": true, "default": null }, { "name": "window", "type": "Tensor", "optional": true, "default": null }, { "name": "normalized", "type": "boolean", "default": false }, { "name": "onesided", "type": "boolean", "optional": true, "default": null }, { "name": "return_complex", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.istft", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "n_fft", "type": "int64" }, { "name": "hop_length", "type": "int64", "optional": true, "default": null }, { "name": "win_length", "type": "int64", "optional": true, "default": null }, { "name": "window", "type": "Tensor", "optional": true, "default": null }, { "name": "center", "type": "boolean", "default": true }, { "name": "normalized", "type": "boolean", "default": false }, { "name": "onesided", "type": "boolean", "optional": true, "default": null }, { "name": "length", "type": "boolean", "optional": true, "default": null }, { "name": "return_complex", "type": "boolean", "default": false } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.unbind:int", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.unbind:Dimname", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "dim", "type": "Dimname" } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.as_tensor", "inputs": [ { "name": "data" }, { "name": "dtype" }, { "name": "device" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.as_strided", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "size", "type": "int64[]" }, { "name": "stride", "type": "int64[]" }, { "name": "storage_offset", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tensor", "inputs": [ { "name": "data" }, { "name": "dtype" }, { "name": "device" }, { "name": "requires_grad", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tensor_split:sections", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "sections", "type": "int64" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.tensor_split:indices", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "indices", "type": "int64[]" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.tensor_split:tensor_indices_or_sections", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "tensor_indices_or_sections", "type": "Tensor" }, { "name": "dim", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor[]" } ] }, { "name": "torch.__upsample:", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "size", "type": "int64", "optional": true, "default": null }, { "name": "scale_factor", "type": "int64", "optional": true, "default": null }, { "name": "mode", "type": "string", "default": "nearest" }, { "name": "align_corners", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__upsample:size_list", "category": "Layer", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "size", "type": "int64[]", "optional": true, "default": null }, { "name": "scale_factor", "type": "int64", "optional": true, "default": null }, { "name": "mode", "type": "string", "default": "nearest" }, { "name": "align_corners", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.triu_", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "diagonal", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.triu:out", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "diagonal", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tril", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "diagonal", "type": "int64", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.tril_indices", "category": "Layer", "inputs": [ { "name": "row", "type": "int64" }, { "name": "col", "type": "int64" }, { "name": "offset", "type": "int64", "default": 0 }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.view_as_real", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.view_as_complex", "category": "Layer", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.torchvision.nms", "inputs": [ { "name": "dets", "type": "Tensor" }, { "name": "scores", "type": "Tensor" }, { "name": "iou_threshold", "type": "float32" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "ops.torchvision.roi_align", "inputs": [ { "name": "input", "type": "Tensor" }, { "name": "rois", "type": "Tensor" }, { "name": "spatial_scale", "type": "float32" }, { "name": "pooled_height", "type": "int64" }, { "name": "pooled_width", "type": "int64" }, { "name": "sampling_ratio", "type": "int64" }, { "name": "aligned", "type": "boolean" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__xor__:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__xor__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__lshift__:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__lshift__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__rshift__:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__rshift__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__ixor__:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__ixor__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__ilshift__:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__ilshift__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__irshift__:Scalar", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Scalar" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.__irshift__:Tensor", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.one_hot:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "num_classes", "type": "int64", "default": -1 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hann_window:", "inputs": [ { "name": "window_length", "type": "int64" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.hann_window:periodic", "inputs": [ { "name": "window_length", "type": "int64" }, { "name": "periodic", "type": "boolean" }, { "name": "dtype", "type": "ScalarType", "optional": true, "default": null }, { "name": "layout", "type": "Layout", "optional": true, "default": null }, { "name": "device", "type": "Device", "optional": true, "default": null }, { "name": "pin_memory", "type": "boolean", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.nonzero_numpy", "inputs": [ { "name": "self", "type": "Tensor" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch._dim_arange:", "inputs": [ { "name": "like", "type": "Tensor" }, { "name": "dim", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cross:", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "dim", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cross:out", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "other", "type": "Tensor" }, { "name": "dim", "type": "int64", "optional": true, "default": null } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.cross_entropy_loss", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "target", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true, "default": null }, { "name": "reduction", "type": "int64" }, { "name": "ignore_index", "type": "int64", "default": -100 }, { "name": "label_smoothing", "type": "float32", "default": 0 } ], "outputs": [ { "name": "output", "type": "Tensor" } ] }, { "name": "torch.binary_cross_entropy_with_logits", "inputs": [ { "name": "self", "type": "Tensor" }, { "name": "target", "type": "Tensor" }, { "name": "weight", "type": "Tensor", "optional": true, "default": null }, { "name": "pos_weight", "type": "Tensor", "optional": true, "default": null }, { "name": "reduction", "type": "int64" } ], "outputs": [ { "name": "output", "type": "Tensor" } ] } ]