Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/python/
H A DMlirHloModule.cc109 int64_t indexVectorDim, MlirContext ctx) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/CAPI/
H A DAttributes.cc26 const int64_t *insertedWindowDims, intptr_t nScatteredDimsToOperandDims, in mlirMhloScatterDimensionNumbersGet()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp5269 auto insertedWindowDims = to_vector(dimNumbers.getInsertedWindowDims()); in validateScatterDimensionNumbers() local
5443 auto insertedWindowDims = in verify() local
5868 SmallVector<int64_t> insertedWindowDims; in parse() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc7851 auto insertedWindowDims = to_vector(dimNumbers.getInsertedWindowDims()); in validateScatterDimensionNumbers() local
8025 auto insertedWindowDims = in verify() local
8224 auto insertedWindowDims = in fold() local
8694 SmallVector<int64_t> insertedWindowDims; in parse() local