Searched refs:stitch_info (Results 1 – 4 of 4) sorted by relevance
388 def __init__(self, name, ops, stitch_info=None, recompute_ops=None): argument393 self.stitch_info = stitch_info452 if self.stitch_info:453 if self.stitch_info.stitch_ops:454 lines.append(' stitch -> ' + str(self.stitch_info.stitch_ops))455 if self.stitch_info.stitch_atomic_ops:456 lines.append(' stitch_atomic_ops-> ' + str(self.stitch_info.stitch_atomic_ops))499 if self.stitch_info and self.stitch_info.stitch_ops:500 buffer_stitch = {'stitch_op': list(self.stitch_info.stitch_ops)}501 if self.stitch_info.stitch_atomic_ops:[all …]
174 if subgraph.stitch_info and subgraph.stitch_info.stitch_ops:175 buffer_stitch = {'stitch_op': list(subgraph.stitch_info.stitch_ops)}176 if subgraph.stitch_info.stitch_atomic_ops:177 buffer_stitch['stitch_atomic_op'] = list(subgraph.stitch_info.stitch_atomic_ops)
142 self.stitch_info = self.StitchInfo()192 def update_stitch_info(self, stitch_info): argument194 if stitch_info.stitch_ops:195 self.stitch_info.stitch_ops.update(stitch_info.stitch_ops)196 if stitch_info.stitch_atomic_ops:197 self.stitch_info.stitch_atomic_ops.update(stitch_info.stitch_atomic_ops)243 self.update_stitch_info(area.stitch_info)464 …subgraphs.append(Graph('{}_{}'.format(self.graph.name, i), area.ops, area.stitch_info, area.recomp…833 dom.stitch_info.stitch_ops.add(dom.ops[0].output.name)836 dom.stitch_info.stitch_ops.add(dom.ops[0].output.name)[all …]
64 if len(subgraphs) > 1 or subgraphs[0].stitch_info.has_stitch_op():