Searched defs:DummyState (Results 1 – 1 of 1) sorted by relevance
71 class DummyState: class80 def dummy_hook_for_no_shard_fsdp(self, state: DummyState, grad: torch.Tensor):99 self, state: DummyState, grad: torch.Tensor, output: torch.Tensor