Home
last modified time | relevance | path

Searched defs:startIndexMap (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.cc211 auto startIndexMap = op.dimension_numbers().getStartIndexMap(); in matchAndRewrite() local
H A Dlegalize_to_linalg.cc2985 ArrayRef<int64_t> startIndexMap = 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.cc273 auto startIndexMap = gatherDim.getStartIndexMap(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/CAPI/
H A DAttributes.cc98 intptr_t nStartIndexMap, const int64_t *startIndexMap, in mlirMhloGatherDimensionNumbersGet()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp1351 auto startIndexMap = to_vector(dimensionNumbers.getStartIndexMap()); in verifyGather() local
5903 SmallVector<int64_t> startIndexMap; in parse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc1463 auto startIndexMap = to_vector(dimensionNumbers.getStartIndexMap()); in verifyGather() local
8729 SmallVector<int64_t> startIndexMap; in parse() local