Searched refs:unwrap_tensor_subclasses (Results 1 – 4 of 4) sorted by relevance
62 unwrap_tensor_subclasses,779 unwrapped_outs = unwrap_tensor_subclasses(797 args_unwrapped = unwrap_tensor_subclasses(
56 unwrap_tensor_subclasses,629 unwrapped_args = unwrap_tensor_subclasses(1849 unwrap_tensor_subclasses(1868 all_args = unwrap_tensor_subclasses(
114 def unwrap_tensor_subclasses(wrapped_args, *, is_joint_structure: bool): function
101 unwrap_tensor_subclasses,