Searched refs:dom_op (Results 1 – 2 of 2) sorted by relevance
255 def dom_op(self): member in GraphSplitByPattern.Area296 if ori == area.dom_op():715 a.dom_op().output.shape != dom.dom_op().output.shape:725 a.dom_op().output.shape == dom.dom_op().output.shape:750 (fused and fused[0].dom_op().output.shape != a.dom_op().output.shape):850 if dom.dom_op().prim != "StridedSlice":861 if not dom.dom_op().prim in gather_prims:913 arg_idx = fuse_arg.get(dom.dom_op().prim, -1)916 fuse_tensor = dom.dom_op().inputs[arg_idx]983 op = dom.dom_op()[all …]
44 self.dom_op = [out.op for out in outputs]63 const_size = max([self.prod(op.output.shape) for op in self.dom_op])117 space = max([_cal_default_space(op) for op in self.dom_op])126 def _ops_type(ops, dom_op): argument131 return PrimLib.iter_type(dom_op[0])133 dom_type = _ops_type(self.ops, self.dom_op)