Searched refs:GetNParallelLoopsAttrs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 68 SmallVector<StringRef, 3> GetNParallelLoopsAttrs(unsigned nParallelLoops) { in GetNParallelLoopsAttrs() function
585 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
725 indexing_maps, GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
883 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
926 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
937 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
1250 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
1323 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
2531 /*outputs=*/init_op, indexing_maps, GetNParallelLoopsAttrs(rank)); in matchAndRewrite()
2645 GetNParallelLoopsAttrs(nloops), in matchAndRewrite()
|