Searched refs:doesnt_support_saved_tensors_hooks (Results 1 – 2 of 2) sorted by relevance
46 from .vmap import doesnt_support_saved_tensors_hooks, get_chunk_sizes371 @doesnt_support_saved_tensors_hooks1396 @doesnt_support_saved_tensors_hooks
39 def doesnt_support_saved_tensors_hooks(f): function