Searched refs:generic_op (Results 1 – 2 of 2) sorted by relevance
151 func.walk([&](linalg::GenericOp generic_op) { in runOnFunction() argument155 tile_sizes = SmallVector<int64_t, 2>(generic_op.getNumLoops(), 1); in runOnFunction()157 auto op = cast<LinalgOp>(generic_op.getOperation()); in runOnFunction()161 generic_op.erase(); in runOnFunction()
295 if (auto generic_op = dyn_cast<linalg::GenericOp>(op)) { in can_reuse_locally() local305 auto operand_buffers = generic_op.getShapedOperands(); in can_reuse_locally()310 AffineMap old_indexing_map = generic_op.getIndexingMap(old_index); in can_reuse_locally()311 AffineMap new_indexing_map = generic_op.getIndexingMap(new_index); in can_reuse_locally()