/usr/local/lib64/python3.6/site-packages/torch/jit
NameSizeModeActions
mobile/-0755rm
__pycache__/-0755rm
annotations.py161260644editdlrm
frontend.py405840644editdlrm
quantized.py238160644editdlrm
supported_ops.py103540644editdlrm
unsupported_tensor_ops.py18300644editdlrm
_async.py37870644editdlrm
_builtins.py62050644editdlrm
_check.py92800644editdlrm
_freeze.py90470644editdlrm
_fuser.py38190644editdlrm
_logging.py2560644editdlrm
_monkeytype_config.py71290644editdlrm
_pickle.py9540644editdlrm
_recursive.py400580644editdlrm
_script.py608250644editdlrm
_serialization.py62690644editdlrm
_state.py36510644editdlrm
_trace.py458730644editdlrm
__init__.py64630644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/jit/unsupported_tensor_ops.py (1830B)
import torch.jit from textwrap import dedent from typing import Dict, Any def execWrapper(code, glob, loc): exec(code, glob, loc) def _gen_unsupported_methods_properties(): tensor_attrs = set(filter(lambda x: x[0] != "_", dir(torch.Tensor))) tensor = torch.tensor([2]) funcs_template = dedent(''' def func(x): return x.{op}() ''') deprecated_apis = set(["volatile", "resize", "reinforce", "new", "name", "map2_", "has_names", "grad_fn", "resize_as"]) tensor_attrs = tensor_attrs - deprecated_apis properties = [] methods = [] sorted_tensor_attrs = sorted(list(tensor_attrs), key=lambda x: x.lower()) for attr in sorted_tensor_attrs: funcs_str = funcs_template.format(op=attr) scope: Dict[str, Any] = {} execWrapper(funcs_str, globals(), scope) try: cu = torch.jit.CompilationUnit(funcs_str) except Exception as e: if "nonexistent attribute" not in repr(e): continue attr_repr = repr(getattr(tensor, attr)) if "bound method" in attr_repr or "built-in method" in attr_repr: methods.append(attr) else: properties.append(attr) mapped_methods = ("\t* :meth:`~torch.Tensor." + x + r"`" for x in methods) mapped_properties = ("\t* :attr:`~torch.Tensor." + x + r"`" for x in properties) return "\n".join(mapped_methods), "\n".join(mapped_properties) def _list_unsupported_tensor_ops(): header = """\n\n Unsupported Tensor Methods ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ """ methods, properties = _gen_unsupported_methods_properties() return header + "\n" + methods + """ Unsupported Tensor Properties ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ """ + "\n" + properties __doc__ = _list_unsupported_tensor_ops()