o Zh@sNddlZejddZedejedddejded ejfd d ZdS) NZexportZFRAGMENTzLaccess_subclass_inner_tensor(Tensor src_subclass_tensor, str attr) -> TensorZaccess_subclass_inner_tensorZAutogradsrc_subclass_tensorattrreturncCsPddlm}||s Jt||d}|dust|tjs&td|d||S)Nr)is_traceable_wrapper_subclassz Attribute z% is not a tensor or doesn't exist in )Ztorch.utils._python_dispatchrgetattr isinstancetorchTensor RuntimeError)rrrvalr F/var/www/auris/lib/python3.10/site-packages/torch/export/custom_ops.py_access_subclass_inner_tensor s   r) rZlibraryLibrarylibZdefineimplr strrr r r r s