Home
last modified time | relevance | path

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

/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.cc272 auto offsetDims = gatherDim.getOffsetDims(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/CAPI/
H A DAttributes.cc96 MlirContext ctx, intptr_t nOffsetDims, const int64_t *offsetDims, in mlirMhloGatherDimensionNumbersGet()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1394 auto offsetDims = to_vector(dimensionNumbers.getOffsetDims()); in verifyGather() local
1521 ArrayRef<int64_t> offsetDims = dimensionNumbers.getOffsetDims(); in inferGatherShape() local
1568 ArrayRef<int64_t> offsetDims = dimensionNumbers.getOffsetDims(); in inferGatherReturnTypeComponents() local
5901 SmallVector<int64_t> offsetDims; in parse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1506 auto offsetDims = to_vector(dimensionNumbers.getOffsetDims()); in verifyGather() local
1633 ArrayRef<int64_t> offsetDims = dimensionNumbers.getOffsetDims(); in inferGatherShape() local
1680 ArrayRef<int64_t> offsetDims = dimensionNumbers.getOffsetDims(); in inferGatherReturnTypeComponents() local
8727 SmallVector<int64_t> offsetDims; in parse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dlegalize_to_linalg.cc2982 ArrayRef<int64_t> offsetDims = gatherOp.dimension_numbers().getOffsetDims(); in matchAndRewrite() local