Searched refs:initTensorOp (Results 1 – 1 of 1) sorted by relevance
1357 auto initTensorOp = dimOp.getSource().getDefiningOp<linalg::InitTensorOp>(); in matchAndRewrite() local1358 if (!initTensorOp || !maybeConstantIndex) in matchAndRewrite()1360 if (!initTensorOp.isDynamicSize(*maybeConstantIndex)) in matchAndRewrite()1362 rewriter.replaceOp(dimOp, initTensorOp.getDynamicSize(*maybeConstantIndex)); in matchAndRewrite()