Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_mhlo_to_thlo.cc206 auto collapsedSliceDims = op.dimension_numbers().getCollapsedSliceDims(); in matchAndRewrite() local
H A Dlegalize_to_linalg.cc2983 ArrayRef<int64_t> collapsedSliceDims = in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/
H A DMlirHloModule.cc149 MlirContext ctx) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_legalize_to_affine.cc271 auto collapsedSliceDims = gatherDim.getCollapsedSliceDims(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/CAPI/
H A DAttributes.cc97 intptr_t nCollapsedSliceDims, const int64_t *collapsedSliceDims, in mlirMhloGatherDimensionNumbersGet()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1403 auto collapsedSliceDims = to_vector(dimensionNumbers.getCollapsedSliceDims()); in verifyGather() local
1496 ArrayRef<int64_t> collapsedSliceDims = in inferGatherShape() local
5902 SmallVector<int64_t> collapsedSliceDims; in parse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1384 auto collapsedSliceDims = dnums.getCollapsedSliceDims(); in matchAndRewrite() local
1515 auto collapsedSliceDims = to_vector(dimensionNumbers.getCollapsedSliceDims()); in verifyGather() local
1608 ArrayRef<int64_t> collapsedSliceDims = in inferGatherShape() local
8728 SmallVector<int64_t> collapsedSliceDims; in parse() local