Searched refs:tile_sizes_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_fuse_linalg.cc | 49 tile_sizes_ = tile_sizes; in LhloFuseLinalgPass()
152 SmallVector<int64_t, 2> tile_sizes(tile_sizes_.begin(), in runOnFunction()
153 tile_sizes_.end()); in runOnFunction()
211 ListOption<unsigned> tile_sizes_{ member in mlir::lmhlo::__anon55bd11270111::LhloFuseLinalgPass
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | kernel_mapping_scheme.h | 95 tile_sizes_{tile_sizes[0], tile_sizes[1], tile_sizes[2]},
123 int64 GetTileSizeFor(int d) const { return tile_sizes_.at(d); } in GetTileSizeFor()
145 const std::array<int64, 3> tile_sizes_; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 154 tile_sizes_ = llvm::to_vector<4>(tile_sizes); in TileLoops()
155 outer_tile_ = tile_sizes_; in TileLoops()
175 tileParallelLoop(ploop, tile_sizes_); in runOnFunction()
193 llvm::SmallVector<int64_t, 4> tile_sizes_; member in tensorflow::kernel_gen::__anon448e82ef0111::TileLoops
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | lmhlo_passes.td | 30 ListOption<"tile_sizes_", "tile-sizes", "unsigned",
|