Home
last modified time | relevance | path

Searched refs:getParallelAndReductionIterators (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg_utils.cc49 SmallVector<StringRef, 3> getParallelAndReductionIterators( in getParallelAndReductionIterators() function
58 return getParallelAndReductionIterators(nParallelLoops, 0); in getNParallelLoopsAttrs()
H A Dlegalize_to_linalg.cc241 getParallelAndReductionIterators(/*nLoops=*/targetRank, in matchAndRewrite()
1664 getParallelAndReductionIterators(srcRank, reductionDims.size()), in matchAndRewrite()
2589 getParallelAndReductionIterators(rank + filteredWindowDims.size(), in matchAndRewrite()
3223 getParallelAndReductionIterators( in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg_utils.h52 SmallVector<StringRef, 3> getParallelAndReductionIterators(unsigned nLoops,