Searched refs:op_sizes (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | graph_parallel.py | 210 def pipeline_fusion_analyze(blocks, op_sizes, exclude_id): argument
220 for b, s in zip(blocks, op_sizes)]
255 def fusion_consult(blocks, op_sizes, exclude_gid): argument
264 blocks, op_sizes, exclude_gid)
274 sum_block, max_weight, sum_weight, blocks, op_sizes, exclude_gid = 0, 0, 0, [], [], []
283 op_sizes.append(len(s.ops))
289 fusion_type, type_info = ScheduleAnalyzer.fusion_consult(blocks, op_sizes, tuple(exclude_gid))
|