Searched defs:indexVectorDim (Results 1 – 8 of 8) sorted by relevance
109 int64_t indexVectorDim, MlirContext ctx) { in PYBIND11_MODULE()149 MlirContext ctx) { in PYBIND11_MODULE()
47 int64_t indexVectorDim = std::max<int64_t>(0, startIndicesTy.getRank() - 1); in matchAndRewrite() local
195 int indexVectorDim = op.dimension_numbers().getIndexVectorDim(); in matchAndRewrite() local
2980 int64_t indexVectorDim = gatherOp.dimension_numbers().getIndexVectorDim(); in matchAndRewrite() local
27 const int64_t *scatteredDimsToOperandDims, int64_t indexVectorDim) { in mlirMhloScatterDimensionNumbersGet()99 int64_t indexVectorDim) { in mlirMhloGatherDimensionNumbersGet()
1348 int64_t indexVectorDim = dimensionNumbers.getIndexVectorDim(); in verifyGather() local1498 int64_t indexVectorDim = dimensionNumbers.getIndexVectorDim(); in inferGatherShape() local5305 auto indexVectorDim = dimNumbers.getIndexVectorDim(); in validateScatterDimensionNumbers() local5373 int64_t indexVectorDim = scatter_dimension_numbers().getIndexVectorDim(); in verify() local5870 int64_t indexVectorDim = 0; in parse() local5904 int64_t indexVectorDim = 0; in parse() local
274 int64_t indexVectorDim = gatherDim.getIndexVectorDim(); in matchAndRewrite() local
1460 int64_t indexVectorDim = dimensionNumbers.getIndexVectorDim(); in verifyGather() local1610 int64_t indexVectorDim = dimensionNumbers.getIndexVectorDim(); in inferGatherShape() local7887 auto indexVectorDim = dimNumbers.getIndexVectorDim(); in validateScatterDimensionNumbers() local7955 int64_t indexVectorDim = scatter_dimension_numbers().getIndexVectorDim(); in verify() local8166 const int64_t indexVectorDim = in fold() local8696 int64_t indexVectorDim = 0; in parse() local8730 int64_t indexVectorDim = 0; in parse() local