Home
last modified time | relevance | path

Searched defs:lambda_fn (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_wrap.py517 def lambda_fn(module: nn.Module): function
528 def lambda_fn(module: nn.Module): function
/aosp_15_r20/external/pytorch/test/distributed/_composable/fully_shard/
H A Dtest_fully_shard_init.py49 def lambda_fn(module: nn.Module): function