wrap_tensor_subclasses_maybe_joint (reference) in projects: aosp_15_r20

Searched refs:wrap_tensor_subclasses_maybe_joint (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A D	traced_function_transforms.py	`63 wrap_tensor_subclasses_maybe_joint, 762 all_args = wrap_tensor_subclasses_maybe_joint(`
H A D	subclass_utils.py	`221 def wrap_tensor_subclasses_maybe_joint( function`
/aosp_15_r20/external/pytorch/torch/_functorch/
H A D	aot_autograd.py	`103 wrap_tensor_subclasses_maybe_joint,`