Searched refs:TileLoop (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_transforms.h | 91 mlir::AffineForOp TileLoop(mlir::AffineForOp loop, int64_t size,
|
| D | conv_emitter.cc | 425 TileLoop(reduction_loops[0], 4, reduction_loops.back()); in TransformMlirConv()
429 TileLoop(cartesian_product_loops[1], 32, cartesian_product_loops.back())); in TransformMlirConv()
431 tiled_cartesian_loops.push_back(TileLoop(cartesian_product_loops.back(), 16, in TransformMlirConv()
|
| D | conv_emitter_transforms.cc | 88 mlir::AffineForOp TileLoop(mlir::AffineForOp loop, int64_t size, in TileLoop() function
|