Searched refs:squeeze_dim (Results 1 – 3 of 3) sorted by relevance
1424 auto squeeze_dim = rewriter.getI64ArrayAttr({-2}); in matchAndRewrite() local1426 rfft_op.getLoc(), output_type, rfft2d.getResult(), squeeze_dim); in matchAndRewrite()
751 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local752 squeeze_dims.emplace_back(squeeze_dim.dyn_cast<IntegerAttr>().getInt()); in matchAndRewrite()
1669 for (auto& squeeze_dim : squeeze_dims_attr) { in matchAndRewrite() local1670 squeeze_dims.emplace_back(squeeze_dim.dyn_cast<IntegerAttr>().getInt()); in matchAndRewrite()