Searched defs:_partitioner (Results 1 – 3 of 3) sorted by relevance
109 def _partitioner(shape, dtype): function179 def _partitioner(shape, dtype): function268 def _partitioner(shape, **unused_args): function
894 def _partitioner(shape, dtype): # pylint:disable=unused-argument function1020 def _partitioner(shape, dtype): # pylint:disable=unused-argument function1080 def _partitioner(shape, dtype): # pylint:disable=unused-argument function1150 def _partitioner(shape, dtype): # pylint:disable=unused-argument function
351 def _partitioner(gm: torch.fx.GraphModule) -> torch.fx.GraphModule: function