Searched defs:_prepare_backward_infra (Results 1 – 2 of 2) sorted by relevance
56 def _prepare_backward_infra(self, n_microbatches): member in MockPipelineStage
257 def _prepare_backward_infra(self, num_microbatches: int): member in _PipelineStageBase