Lines Matching refs:getDefiningOp
102 auto producer = cast<GenericOp>(consumerOpOperand->get().getDefiningOp()); in generateFusedElementwiseOpRegion()
194 assert(!producer->isAncestor(replacement.getDefiningOp()) && in generateFusedElementwiseOpRegion()
820 auto reshapeOp = en.value()->get().getDefiningOp<TensorReshapeOp>(); in matchAndRewrite()
936 en.value()->get().template getDefiningOp<TensorExpandShapeOp>(); in matchAndRewrite()
1044 opOperand->get().getDefiningOp<TensorCollapseShapeOp>(); in matchAndRewrite()
1078 GenericOp producer = reshapeOp.src().template getDefiningOp<GenericOp>(); in matchAndRewrite()
1140 GenericOp producer = reshapeOp.src().getDefiningOp<GenericOp>(); in matchAndRewrite()
1167 Operation *def = opOperand->get().getDefiningOp(); in matchAndRewrite()
1245 auto expandShapeOp = producer.getDefiningOp<linalg::TensorExpandShapeOp>(); in skipUnitDimReshape()
1249 producer.getDefiningOp<linalg::TensorCollapseShapeOp>(); in skipUnitDimReshape()
1266 dyn_cast_or_null<GenericOp>(opOperand->get().getDefiningOp()); in matchAndRewrite()
1336 auto definingOp = operandVal.getDefiningOp<InitTensorOp>(); in matchAndRewrite()