Searched refs:getNParallelLoopsAttrs (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| H A D | legalize_to_linalg_utils.h | 56 SmallVector<StringRef, 3> getNParallelLoopsAttrs(unsigned nParallelLoops);
176 getNParallelLoopsAttrs(nloops), in matchAndRewrite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | legalize_to_linalg_utils.cc | 57 SmallVector<StringRef, 3> getNParallelLoopsAttrs(unsigned nParallelLoops) { in getNParallelLoopsAttrs() function
|
| H A D | legalize_to_linalg.cc | 626 indexingMaps, getNParallelLoopsAttrs(nloops), in matchAndRewrite()
790 getNParallelLoopsAttrs(nloops), in matchAndRewrite()
1058 getNParallelLoopsAttrs(nloops), in matchAndRewrite()
1115 getNParallelLoopsAttrs(nloops), in matchAndRewrite()
1529 getNParallelLoopsAttrs(resultType.getRank()), in matchAndRewrite()
2908 /*outputs=*/initOp, indexingMaps, getNParallelLoopsAttrs(rank), in matchAndRewrite()
3028 /*outputs=*/initOp, indexingMaps, getNParallelLoopsAttrs(resultRank), in matchAndRewrite()
|