mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-07 08:04:18 +00:00
1 line
100 KiB
JSON
1 line
100 KiB
JSON
{".class": "MypyFile", "_fullname": "torch.nn.functional", "is_partial_stub_package": false, "is_stub": true, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Callable": {".class": "SymbolTableNode", "cross_ref": "typing.Callable", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Dict": {".class": "SymbolTableNode", "cross_ref": "typing.Dict", "kind": "Gdef", "module_hidden": true, "module_public": false}, "GRID_SAMPLE_INTERPOLATION_MODES": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.nn.functional.GRID_SAMPLE_INTERPOLATION_MODES", "line": 15, "no_args": false, "normalized": false, "target": {".class": "Instance", "args": ["builtins.str", "builtins.int"], "type_ref": "builtins.dict"}}}, "GRID_SAMPLE_PADDING_MODES": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.nn.functional.GRID_SAMPLE_PADDING_MODES", "line": 16, "no_args": false, "normalized": false, "target": {".class": "Instance", "args": ["builtins.str", "builtins.int"], "type_ref": "builtins.dict"}}}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Sequence": {".class": "SymbolTableNode", "cross_ref": "typing.Sequence", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Tuple": {".class": "SymbolTableNode", "cross_ref": "typing.Tuple", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Union": {".class": "SymbolTableNode", "cross_ref": "typing.Union", "kind": "Gdef", "module_hidden": true, "module_public": false}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.__package__", "name": "__package__", "type": "builtins.str"}}, "_ratio_any_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._ratio_any_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size": {".class": "SymbolTableNode", "cross_ref": "torch.types._size", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size_1_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._size_1_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size_2_opt_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._size_2_opt_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size_2_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._size_2_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size_3_opt_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._size_3_opt_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size_3_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._size_3_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_size_any_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.common_types._size_any_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "adaptive_avg_pool1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "flags": [], "fullname": "torch.nn.functional.adaptive_avg_pool1d", "name": "adaptive_avg_pool1d", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_1_t"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_avg_pool1d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "adaptive_avg_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "flags": [], "fullname": "torch.nn.functional.adaptive_avg_pool2d", "name": "adaptive_avg_pool2d", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_2_opt_t"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_avg_pool2d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "adaptive_avg_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "flags": [], "fullname": "torch.nn.functional.adaptive_avg_pool3d", "name": "adaptive_avg_pool3d", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_3_opt_t"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_avg_pool3d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "adaptive_max_pool1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.adaptive_max_pool1d", "name": "adaptive_max_pool1d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "adaptive_max_pool1d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["input", "output_size", "return_indices"], "flags": [], "fullname": "torch.nn.functional.adaptive_max_pool1d_with_indices", "name": "adaptive_max_pool1d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["input", "output_size", "return_indices"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_max_pool1d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "adaptive_max_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.adaptive_max_pool2d", "name": "adaptive_max_pool2d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "adaptive_max_pool2d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["input", "output_size", "return_indices"], "flags": [], "fullname": "torch.nn.functional.adaptive_max_pool2d_with_indices", "name": "adaptive_max_pool2d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["input", "output_size", "return_indices"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_2_opt_t"}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_max_pool2d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "adaptive_max_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.adaptive_max_pool3d", "name": "adaptive_max_pool3d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "adaptive_max_pool3d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["input", "output_size", "return_indices"], "flags": [], "fullname": "torch.nn.functional.adaptive_max_pool3d_with_indices", "name": "adaptive_max_pool3d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["input", "output_size", "return_indices"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_3_opt_t"}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_max_pool3d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "affine_grid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["theta", "size", "align_corners"], "flags": [], "fullname": "torch.nn.functional.affine_grid", "name": "affine_grid", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["theta", "size", "align_corners"], "arg_types": ["torch._tensor.Tensor", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "affine_grid", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "alpha_dropout": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "flags": [], "fullname": "torch.nn.functional.alpha_dropout", "name": "alpha_dropout", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "alpha_dropout", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "assert_int_or_pair": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["arg", "arg_name", "message"], "flags": [], "fullname": "torch.nn.functional.assert_int_or_pair", "name": "assert_int_or_pair", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["arg", "arg_name", "message"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "assert_int_or_pair", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "avg_pool1d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.avg_pool1d", "kind": "Gdef"}, "avg_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.avg_pool2d", "name": "avg_pool2d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "avg_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.avg_pool3d", "name": "avg_pool3d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "batch_norm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "running_mean", "running_var", "weight", "bias", "training", "momentum", "eps"], "flags": [], "fullname": "torch.nn.functional.batch_norm", "name": "batch_norm", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "running_mean", "running_var", "weight", "bias", "training", "momentum", "eps"], "arg_types": ["torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", "builtins.float", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "batch_norm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "bilinear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input1", "input2", "weight", "bias"], "flags": [], "fullname": "torch.nn.functional.bilinear", "name": "bilinear", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input1", "input2", "weight", "bias"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "bilinear", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "binary_cross_entropy": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.binary_cross_entropy", "name": "binary_cross_entropy", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "binary_cross_entropy", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "binary_cross_entropy_with_logits": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "reduce", "reduction", "pos_weight"], "flags": [], "fullname": "torch.nn.functional.binary_cross_entropy_with_logits", "name": "binary_cross_entropy_with_logits", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "reduce", "reduction", "pos_weight"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "binary_cross_entropy_with_logits", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "celu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "alpha", "inplace"], "flags": [], "fullname": "torch.nn.functional.celu", "name": "celu", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1], "arg_names": ["input", "alpha", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "celu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "celu_": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.celu_", "kind": "Gdef"}, "channel_shuffle": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.channel_shuffle", "kind": "Gdef"}, "conv1d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv1d", "kind": "Gdef"}, "conv2d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv2d", "kind": "Gdef"}, "conv3d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv3d", "kind": "Gdef"}, "conv_tbc": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv_tbc", "kind": "Gdef"}, "conv_transpose1d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv_transpose1d", "kind": "Gdef"}, "conv_transpose2d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv_transpose2d", "kind": "Gdef"}, "conv_transpose3d": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.conv_transpose3d", "kind": "Gdef"}, "cosine_embedding_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1], "arg_names": ["input1", "input2", "target", "margin", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.cosine_embedding_loss", "name": "cosine_embedding_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1], "arg_names": ["input1", "input2", "target", "margin", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.float", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "cosine_embedding_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "cosine_similarity": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.cosine_similarity", "kind": "Gdef"}, "cross_entropy": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "ignore_index", "reduce", "reduction", "label_smoothing"], "flags": [], "fullname": "torch.nn.functional.cross_entropy", "name": "cross_entropy", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "ignore_index", "reduce", "reduction", "label_smoothing"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.int", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "cross_entropy", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "ctc_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 1, 1, 1], "arg_names": ["log_probs", "targets", "input_lengths", "target_lengths", "blank", "reduction", "zero_infinity"], "flags": [], "fullname": "torch.nn.functional.ctc_loss", "name": "ctc_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 1, 1, 1], "arg_names": ["log_probs", "targets", "input_lengths", "target_lengths", "blank", "reduction", "zero_infinity"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.int", "builtins.str", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "ctc_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "dropout": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "flags": [], "fullname": "torch.nn.functional.dropout", "name": "dropout", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "dropout", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "dropout2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "flags": [], "fullname": "torch.nn.functional.dropout2d", "name": "dropout2d", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "dropout2d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "dropout3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "flags": [], "fullname": "torch.nn.functional.dropout3d", "name": "dropout3d", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "dropout3d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "elu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "alpha", "inplace"], "flags": [], "fullname": "torch.nn.functional.elu", "name": "elu", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1], "arg_names": ["input", "alpha", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "elu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "elu_": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.elu_", "name": "elu_", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "embedding": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "padding_idx", "max_norm", "norm_type", "scale_grad_by_freq", "sparse"], "flags": [], "fullname": "torch.nn.functional.embedding", "name": "embedding", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "padding_idx", "max_norm", "norm_type", "scale_grad_by_freq", "sparse"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "embedding", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "embedding_bag": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "offsets", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "per_sample_weights", "include_last_offset", "padding_idx"], "flags": [], "fullname": "torch.nn.functional.embedding_bag", "name": "embedding_bag", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "offsets", "max_norm", "norm_type", "scale_grad_by_freq", "mode", "sparse", "per_sample_weights", "include_last_offset", "padding_idx"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.str", "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "embedding_bag", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "feature_alpha_dropout": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "flags": [], "fullname": "torch.nn.functional.feature_alpha_dropout", "name": "feature_alpha_dropout", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "p", "training", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "feature_alpha_dropout", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "fold": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "output_size", "kernel_size", "dilation", "padding", "stride"], "flags": [], "fullname": "torch.nn.functional.fold", "name": "fold", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "output_size", "kernel_size", "dilation", "padding", "stride"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "fold", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "fractional_max_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.fractional_max_pool2d", "name": "fractional_max_pool2d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "fractional_max_pool2d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "output_size", "output_ratio", "return_indices", "_random_samples"], "flags": [], "fullname": "torch.nn.functional.fractional_max_pool2d_with_indices", "name": "fractional_max_pool2d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "output_size", "output_ratio", "return_indices", "_random_samples"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "Instance", "args": ["builtins.float"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "fractional_max_pool2d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "fractional_max_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.fractional_max_pool3d", "name": "fractional_max_pool3d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "fractional_max_pool3d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "output_size", "output_ratio", "return_indices", "_random_samples"], "flags": [], "fullname": "torch.nn.functional.fractional_max_pool3d_with_indices", "name": "fractional_max_pool3d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "output_size", "output_ratio", "return_indices", "_random_samples"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "Instance", "args": ["builtins.float"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "fractional_max_pool3d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "gaussian_nll_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "target", "var", "full", "eps", "reduction"], "flags": [], "fullname": "torch.nn.functional.gaussian_nll_loss", "name": "gaussian_nll_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "target", "var", "full", "eps", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.str", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "gaussian_nll_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "gelu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input"], "flags": [], "fullname": "torch.nn.functional.gelu", "name": "gelu", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["input"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "gelu", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "glu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "dim"], "flags": [], "fullname": "torch.nn.functional.glu", "name": "glu", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "dim"], "arg_types": ["torch._tensor.Tensor", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "glu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "grid_sample": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "grid", "mode", "padding_mode", "align_corners"], "flags": [], "fullname": "torch.nn.functional.grid_sample", "name": "grid_sample", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "grid", "mode", "padding_mode", "align_corners"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.str", "builtins.str", {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "grid_sample", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "group_norm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "num_groups", "weight", "bias", "eps"], "flags": [], "fullname": "torch.nn.functional.group_norm", "name": "group_norm", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "num_groups", "weight", "bias", "eps"], "arg_types": ["torch._tensor.Tensor", "builtins.int", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "group_norm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "gumbel_softmax": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["logits", "tau", "hard", "eps", "dim"], "flags": [], "fullname": "torch.nn.functional.gumbel_softmax", "name": "gumbel_softmax", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["logits", "tau", "hard", "eps", "dim"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", "builtins.float", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "gumbel_softmax", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardshrink": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "lambd"], "flags": [], "fullname": "torch.nn.functional.hardshrink", "name": "hardshrink", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "lambd"], "arg_types": ["torch._tensor.Tensor", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardshrink", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardsigmoid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.hardsigmoid", "name": "hardsigmoid", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardsigmoid", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardswish": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.hardswish", "name": "hardswish", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardswish", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardtanh": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "min_val", "max_val", "inplace"], "flags": [], "fullname": "torch.nn.functional.hardtanh", "name": "hardtanh", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "min_val", "max_val", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardtanh", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardtanh_": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.hardtanh_", "name": "hardtanh_", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "hinge_embedding_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "margin", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.hinge_embedding_loss", "name": "hinge_embedding_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "margin", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.float", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hinge_embedding_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "huber_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["input", "target", "reduction", "delta"], "flags": [], "fullname": "torch.nn.functional.huber_loss", "name": "huber_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1], "arg_names": ["input", "target", "reduction", "delta"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.str", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "huber_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "instance_norm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "running_mean", "running_var", "weight", "bias", "use_input_stats", "momentum", "eps"], "flags": [], "fullname": "torch.nn.functional.instance_norm", "name": "instance_norm", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "running_mean", "running_var", "weight", "bias", "use_input_stats", "momentum", "eps"], "arg_types": ["torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", "builtins.float", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "instance_norm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "interpolate": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "size", "scale_factor", "mode", "align_corners"], "flags": [], "fullname": "torch.nn.functional.interpolate", "name": "interpolate", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "size", "scale_factor", "mode", "align_corners"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, "builtins.str", {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "interpolate", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "kl_div": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction", "log_target"], "flags": [], "fullname": "torch.nn.functional.kl_div", "name": "kl_div", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction", "log_target"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "kl_div", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "l1_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.l1_loss", "name": "l1_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "l1_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "layer_norm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "normalized_shape", "weight", "bias", "eps"], "flags": [], "fullname": "torch.nn.functional.layer_norm", "name": "layer_norm", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "normalized_shape", "weight", "bias", "eps"], "arg_types": ["torch._tensor.Tensor", {".class": "Instance", "args": ["builtins.int"], "type_ref": "typing.Sequence"}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "layer_norm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "leaky_relu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "negative_slope", "inplace"], "flags": [], "fullname": "torch.nn.functional.leaky_relu", "name": "leaky_relu", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1], "arg_names": ["input", "negative_slope", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "leaky_relu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "leaky_relu_": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.leaky_relu_", "name": "leaky_relu_", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "linear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["input", "weight", "bias"], "flags": [], "fullname": "torch.nn.functional.linear", "name": "linear", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["input", "weight", "bias"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "linear", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "local_response_norm": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "size", "alpha", "beta", "k"], "flags": [], "fullname": "torch.nn.functional.local_response_norm", "name": "local_response_norm", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "size", "alpha", "beta", "k"], "arg_types": ["torch._tensor.Tensor", "builtins.int", "builtins.float", "builtins.float", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "local_response_norm", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "log_softmax": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "dim", "_stacklevel", "dtype"], "flags": [], "fullname": "torch.nn.functional.log_softmax", "name": "log_softmax", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "dim", "_stacklevel", "dtype"], "arg_types": ["torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}, "builtins.int", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "log_softmax", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "logsigmoid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.logsigmoid", "name": "logsigmoid", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "lp_pool1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1], "arg_names": ["input", "norm_type", "kernel_size", "stride", "ceil_mode"], "flags": [], "fullname": "torch.nn.functional.lp_pool1d", "name": "lp_pool1d", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1], "arg_names": ["input", "norm_type", "kernel_size", "stride", "ceil_mode"], "arg_types": ["torch._tensor.Tensor", "builtins.float", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_1_t"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}, "builtins.int", {".class": "NoneType"}]}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "lp_pool1d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "lp_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1], "arg_names": ["input", "norm_type", "kernel_size", "stride", "ceil_mode"], "flags": [], "fullname": "torch.nn.functional.lp_pool2d", "name": "lp_pool2d", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1], "arg_names": ["input", "norm_type", "kernel_size", "stride", "ceil_mode"], "arg_types": ["torch._tensor.Tensor", "builtins.float", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_2_t"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}, "builtins.int", {".class": "NoneType"}]}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "lp_pool2d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "margin_ranking_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1], "arg_names": ["input1", "input2", "target", "margin", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.margin_ranking_loss", "name": "margin_ranking_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1], "arg_names": ["input1", "input2", "target", "margin", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.float", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "margin_ranking_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "max_pool1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.max_pool1d", "name": "max_pool1d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "max_pool1d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "flags": [], "fullname": "torch.nn.functional.max_pool1d_with_indices", "name": "max_pool1d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_pool1d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "max_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.max_pool2d", "name": "max_pool2d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "max_pool2d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "flags": [], "fullname": "torch.nn.functional.max_pool2d_with_indices", "name": "max_pool2d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_pool2d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "max_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.max_pool3d", "name": "max_pool3d", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "max_pool3d_with_indices": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "flags": [], "fullname": "torch.nn.functional.max_pool3d_with_indices", "name": "max_pool3d_with_indices", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_pool3d_with_indices", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "max_unpool1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "indices", "kernel_size", "stride", "padding", "output_size"], "flags": [], "fullname": "torch.nn.functional.max_unpool1d", "name": "max_unpool1d", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "indices", "kernel_size", "stride", "padding", "output_size"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_unpool1d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "max_unpool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "indices", "kernel_size", "stride", "padding", "output_size"], "flags": [], "fullname": "torch.nn.functional.max_unpool2d", "name": "max_unpool2d", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "indices", "kernel_size", "stride", "padding", "output_size"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_unpool2d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "max_unpool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "indices", "kernel_size", "stride", "padding", "output_size"], "flags": [], "fullname": "torch.nn.functional.max_unpool3d", "name": "max_unpool3d", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["input", "indices", "kernel_size", "stride", "padding", "output_size"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.types._size"}, {".class": "UnionType", "items": ["torch._C.Size", {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.list"}, {".class": "Instance", "args": ["builtins.int"], "type_ref": "builtins.tuple"}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_unpool3d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "mish": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.mish", "name": "mish", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mish", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "mse_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.mse_loss", "name": "mse_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mse_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "multi_head_attention_forward": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["query", "key", "value", "embed_dim_to_check", "num_heads", "in_proj_weight", "in_proj_bias", "bias_k", "bias_v", "add_zero_attn", "dropout_p", "out_proj_weight", "out_proj_bias", "training", "key_padding_mask", "need_weights", "attn_mask", "use_separate_proj_weight", "q_proj_weight", "k_proj_weight", "v_proj_weight", "static_k", "static_v"], "flags": [], "fullname": "torch.nn.functional.multi_head_attention_forward", "name": "multi_head_attention_forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["query", "key", "value", "embed_dim_to_check", "num_heads", "in_proj_weight", "in_proj_bias", "bias_k", "bias_v", "add_zero_attn", "dropout_p", "out_proj_weight", "out_proj_bias", "training", "key_padding_mask", "need_weights", "attn_mask", "use_separate_proj_weight", "q_proj_weight", "k_proj_weight", "v_proj_weight", "static_k", "static_v"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.int", "builtins.int", "torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", "builtins.float", "torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, "builtins.bool", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "multi_head_attention_forward", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "multi_margin_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "p", "margin", "weight", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.multi_margin_loss", "name": "multi_margin_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "p", "margin", "weight", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.int", "builtins.float", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "multi_margin_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "multilabel_margin_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.multilabel_margin_loss", "name": "multilabel_margin_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "multilabel_margin_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "multilabel_soft_margin_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.multilabel_soft_margin_loss", "name": "multilabel_soft_margin_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "multilabel_soft_margin_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "nll_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "ignore_index", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.nll_loss", "name": "nll_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "weight", "size_average", "ignore_index", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.int", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "nll_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "normalize": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "p", "dim", "eps", "out"], "flags": [], "fullname": "torch.nn.functional.normalize", "name": "normalize", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "p", "dim", "eps", "out"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.int", "builtins.float", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "normalize", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "one_hot": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.one_hot", "name": "one_hot", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "pad": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["input", "pad", "mode", "value"], "flags": [], "fullname": "torch.nn.functional.pad", "name": "pad", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1], "arg_names": ["input", "pad", "mode", "value"], "arg_types": ["torch._tensor.Tensor", {".class": "Instance", "args": ["builtins.int"], "type_ref": "typing.Sequence"}, "builtins.str", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "pad", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "pairwise_distance": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["x1", "x2", "p", "eps", "keepdim"], "flags": [], "fullname": "torch.nn.functional.pairwise_distance", "name": "pairwise_distance", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["x1", "x2", "p", "eps", "keepdim"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.float", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "pairwise_distance", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "pdist": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.pdist", "kind": "Gdef"}, "pixel_shuffle": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.pixel_shuffle", "kind": "Gdef"}, "pixel_unshuffle": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.pixel_unshuffle", "kind": "Gdef"}, "poisson_nll_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "log_input", "full", "size_average", "eps", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.poisson_nll_loss", "name": "poisson_nll_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "target", "log_input", "full", "size_average", "eps", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.bool", "builtins.bool", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.float", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "poisson_nll_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "prelu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "weight"], "flags": [], "fullname": "torch.nn.functional.prelu", "name": "prelu", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["input", "weight"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "prelu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "relu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.relu", "name": "relu", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "relu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "relu6": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.relu6", "name": "relu6", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "relu6", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "relu_": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.relu_", "kind": "Gdef"}, "rrelu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "lower", "upper", "training", "inplace"], "flags": [], "fullname": "torch.nn.functional.rrelu", "name": "rrelu", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "lower", "upper", "training", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "rrelu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "rrelu_": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.rrelu_", "kind": "Gdef"}, "selu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.selu", "name": "selu", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "selu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "selu_": {".class": "SymbolTableNode", "cross_ref": "torch._C._VariableFunctions.selu_", "kind": "Gdef"}, "sigmoid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input"], "flags": [], "fullname": "torch.nn.functional.sigmoid", "name": "sigmoid", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["input"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "sigmoid", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "silu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "flags": [], "fullname": "torch.nn.functional.silu", "name": "silu", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["input", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "silu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "smooth_l1_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction", "beta"], "flags": [], "fullname": "torch.nn.functional.smooth_l1_loss", "name": "smooth_l1_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction", "beta"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "smooth_l1_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "soft_margin_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.soft_margin_loss", "name": "soft_margin_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "target", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "soft_margin_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "softmax": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "dim", "_stacklevel", "dtype"], "flags": [], "fullname": "torch.nn.functional.softmax", "name": "softmax", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "dim", "_stacklevel", "dtype"], "arg_types": ["torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}, "builtins.int", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "softmax", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "softmin": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "dim", "_stacklevel", "dtype"], "flags": [], "fullname": "torch.nn.functional.softmin", "name": "softmin", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "dim", "_stacklevel", "dtype"], "arg_types": ["torch._tensor.Tensor", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}, "builtins.int", {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "softmin", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "softplus": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.softplus", "name": "softplus", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "softshrink": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.functional.softshrink", "name": "softshrink", "type": {".class": "CallableType", "arg_kinds": [2, 4], "arg_names": [null, null], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": true, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}, "type_guard": null, "variables": []}}}, "softsign": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input"], "flags": [], "fullname": "torch.nn.functional.softsign", "name": "softsign", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["input"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "softsign", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "tanh": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input"], "flags": [], "fullname": "torch.nn.functional.tanh", "name": "tanh", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["input"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "tanh", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "tanhshrink": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input"], "flags": [], "fullname": "torch.nn.functional.tanhshrink", "name": "tanhshrink", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["input"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "tanhshrink", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "threshold": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input", "threshold", "value", "inplace"], "flags": [], "fullname": "torch.nn.functional.threshold", "name": "threshold", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input", "threshold", "value", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "threshold", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "triplet_margin_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["anchor", "positive", "negative", "margin", "p", "eps", "swap", "size_average", "reduce", "reduction"], "flags": [], "fullname": "torch.nn.functional.triplet_margin_loss", "name": "triplet_margin_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["anchor", "positive", "negative", "margin", "p", "eps", "swap", "size_average", "reduce", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.float", "builtins.float", "builtins.float", "builtins.bool", {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.bool", {".class": "NoneType"}]}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "triplet_margin_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "triplet_margin_with_distance_loss": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 5, 5, 5, 5], "arg_names": ["anchor", "positive", "negative", "distance_function", "margin", "swap", "reduction"], "flags": [], "fullname": "torch.nn.functional.triplet_margin_with_distance_loss", "name": "triplet_margin_with_distance_loss", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 5, 5, 5, 5], "arg_names": ["anchor", "positive", "negative", "distance_function", "margin", "swap", "reduction"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": [{".class": "CallableType", "arg_kinds": [0, 0], "arg_names": [null, null], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": null, "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}, {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "triplet_margin_with_distance_loss", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "unfold": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "kernel_size", "dilation", "padding", "stride"], "flags": [], "fullname": "torch.nn.functional.unfold", "name": "unfold", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "kernel_size", "dilation", "padding", "stride"], "arg_types": ["torch._tensor.Tensor", {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.common_types._size_any_t"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "unfold", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "upsample": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "size", "scale_factor", "mode", "align_corners"], "flags": [], "fullname": "torch.nn.functional.upsample", "name": "upsample", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "size", "scale_factor", "mode", "align_corners"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, "builtins.str", {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "upsample", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "upsample_bilinear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "size", "scale_factor"], "flags": [], "fullname": "torch.nn.functional.upsample_bilinear", "name": "upsample_bilinear", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1], "arg_names": ["input", "size", "scale_factor"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "upsample_bilinear", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "upsample_nearest": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "size", "scale_factor"], "flags": [], "fullname": "torch.nn.functional.upsample_nearest", "name": "upsample_nearest", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1], "arg_names": ["input", "size", "scale_factor"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "upsample_nearest", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/functional.pyi"} |