Home
last modified time | relevance | path

Searched defs:tileSizes (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
H A Dtiling.cc45 ArrayRef<int64_t> tileSizes) { in createTile()
81 ArrayRef<int64_t> tileSizes) { in createTileOrPoint()
189 SmallVector<int64_t> tileSizes; in parseTileSizes() local
221 llvm::Optional<SmallVector<int64_t>> tileSizes = parseTileSizes(istream); in parseNestedTileSizes() local
287 llvm::Optional<SmallVector<SmallVector<int64_t>>> tileSizes; member
297 const SmallVector<SmallVector<int64_t>> &tileSizes) { in createTilingPass()
302 const std::string &tileSizes) { in createTilingPass()
H A Dtransforms.cc173 RewriterBase &b, LinalgOp op, ValueRange tileSizes, in tileLinalgOpImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_fuse_linalg.cc52 llvm::ArrayRef<unsigned> tileSizes) { in LhloFuseLinalgPass()
157 SmallVector<int64_t, 2> tileSizes(tile_sizes_.begin(), tile_sizes_.end()); in runOnOperation() local
200 bool tileGenericOp(LinalgOp op, ArrayRef<int64_t> tileSizes, OpBuilder* b) { in tileGenericOp()
215 bool useParallelLoops, ArrayRef<unsigned> tileSizes) { in createLhloFuseLinalgPass()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dtile_loops_pass.cc50 explicit TileLoopsPass(ArrayRef<int64_t> tileSizes, in TileLoopsPass()
131 ArrayRef<int64_t> tileSizes, ArrayRef<int64_t> unrollFactors) { in createTileLoopsPass()
H A Dhlo_to_gpu_pipeline.cc44 ArrayRef<int64_t> tileSizes, in createHloToGpuPipeline()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/thlo/IR/
H A Dthlo_ops.cc483 SmallVector<Value> tileSizes; in fuse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
H A Dgml_st_ops.cc1664 auto tileSizes = getMixedSizes(tileOp.static_sizes(), tileOp.sizes()); in selectDimsFromSet() local