Searched refs:_add_script_class (Results 1 – 2 of 2) sorted by relevance
69 def _add_script_class(python_class, script_class): function
16 from torch.jit._state import _add_script_class, _get_script_class, _python_cu64 _add_script_class(obj, script_class)