Home
last modified time | relevance | path

Searched defs:orderedReductionDims (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dgroup_reduction_dimensions.cc40 SmallVector<int64_t>& orderedReductionDims, PatternRewriter& rewriter) { in tryLowerToCollapseShape()
94 SmallVector<int64_t> orderedReductionDims, in groupDimensions()
132 SmallVector<int64_t>& orderedReductionDims, bool preferColumnsReductions, in tryLowerTo1DOr2DReduction()
272 SmallVector<int64_t> orderedReductionDims = in matchAndRewrite() local