Searched refs:remove_bookend_non_compute_ops (Results 1 – 2 of 2) sorted by relevance
276 def remove_bookend_non_compute_ops(self, partitions: List[Partition]): member in CapabilityBasedPartitioner
302 partitioner.remove_bookend_non_compute_ops(partitions)