Searched refs:TileLoops (Results 1 – 3 of 3) sorted by relevance
150 class TileLoops : public mlir::PassWrapper<TileLoops, mlir::FunctionPass> { class152 explicit TileLoops(llvm::ArrayRef<int64_t> tile_sizes, in TileLoops() function in tensorflow::kernel_gen::__anon86061cf60111::TileLoops249 std::make_unique<TileLoops>(tile_sizes, unroll_factors)); in LowerTFtoLoops()
146 using TileLoops = std::pair<Loops, Loops>; variable237 TileLoops extractFixedOuterLoops(scf::ForOp rootFOrOp, ArrayRef<int64_t> sizes);
1845 static LogicalResult tryIsolateBands(const TileLoops &tileLoops) { in tryIsolateBands()1862 TileLoops mlir::extractFixedOuterLoops(scf::ForOp rootForOp, in extractFixedOuterLoops()1894 TileLoops tileLoops = std::make_pair(forOps, intraTile); in extractFixedOuterLoops()