Lines Matching defs:linalg

40 #define DEBUG_TYPE "linalg-utils"
45 using namespace mlir::linalg;
115   auto addPattern = m_Op<linalg::YieldOp>(m_Op<arith::AddIOp>(a, b));
123 template struct mlir::linalg::GenerateLoopNest<scf::ForOp>;
124 template struct mlir::linalg::GenerateLoopNest<scf::ParallelOp>;
125 template struct mlir::linalg::GenerateLoopNest<AffineForOp>;
147 namespace linalg {
160               linalg::YieldOp, linalg::IndexOp, AffineApplyOp>(op) ||
264   return b.create<linalg::GenericOp>(
271         b.create<linalg::YieldOp>(loc, args.front());
283     ArrayRef<linalg::ProcInfo> procInfo) {
326     ArrayRef<linalg::ProcInfo> /*procInfo*/) {
375     ArrayRef<linalg::ProcInfo> procInfo,
415                      .drop_while([&](linalg::ProcInfo p) {
493     ArrayRef<linalg::ProcInfo> procInfo) {
516     if (it.value().distributionMethod != linalg::DistributionMethod::None) {
882 } // namespace linalg