(self, graph: BaseGraph, topo_order: List[Operation])
| 196 | Besides, there is a parameter 'limit' to control the size of each block. |
| 197 | """ |
| 198 | def __init__(self, graph: BaseGraph, topo_order: List[Operation]) -> None: |
| 199 | self.graph = graph |
| 200 | self.op_orders = topo_order |
| 201 | self.depth = {} |
| 202 | self.search_engine = SearchableGraph(self.graph) |
| 203 | self.initialize_depth() |
| 204 | |
| 205 | def create_block(self, sp: Operation, ep: Operation) -> TrainableBlock: |
| 206 | if sp == ep: return TrainableBlock(sp=sp, ep=ep, rps=[sp]) |
nothing calls this directly
no test coverage detected