Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_checkpoint/
H A Dcheckpoint_wrapper.py294 lambda_auto_wrap_policy,
300 else partial(lambda_auto_wrap_policy, lambda_fn=check_fn)
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A Dwrap.py279 def lambda_auto_wrap_policy( function
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py32 lambda_auto_wrap_policy,
664 wrap_policy = functools.partial(lambda_auto_wrap_policy, lambda_fn=is_inner)