Searched defs:dimIndex (Results 1 – 9 of 9) sorted by relevance
165 Value dimIndex = b.create<tensor::DimOp>(loc, op.getOperand(0), v); in runOnOperation() local231 Value dimIndex = b.create<tensor::DimOp>(loc, op.getOperand(0), i); in runOnOperation() local
162 for (size_t dimIndex = 0; dimIndex < data.mSubArraySizes.size(); ++dimIndex) in RewriteArrayOfArraySubscriptExpression() local
420 for (unsigned int dimIndex = 0; dimIndex < dimensionCount; ++dimIndex) variable445 for (unsigned int dimIndex = 0; dimIndex < dimensionCount; ++dimIndex) variable
125 SmallVector<Value, 4> dimIndex; in elementalLower() local
83 size_t dimIndex, int64_t dim, OpBuilder* b) { in getStaticOrDynamicDim()
1511 for (int dimIndex = 0; dimIndex <= maxCollapsedDim; ++dimIndex) { in inferGatherShape() local2673 auto dimIndex = dimensions.getValues<int64_t>()[i]; in verify() local2741 auto dimIndex = bcastDimensions.getValues<int64_t>()[i]; in verify() local
7864 for (size_t dimIndex = 0; dimIndex < indices.size(); ++dimIndex) in ForEachOpaque() local7874 for (size_t dimIndex = 0; dimIndex < indices.size(); ++dimIndex) in ForEachOpaque() local
7963 for (size_t dimIndex = 0; dimIndex < indices.size(); ++dimIndex) in ForEachOpaque() local7973 for (size_t dimIndex = 0; dimIndex < indices.size(); ++dimIndex) in ForEachOpaque() local
1623 for (int dimIndex = 0; dimIndex <= maxCollapsedDim; ++dimIndex) { in inferGatherShape() local3220 auto dimIndex = dimensions.getValues<int64_t>()[i]; in verify() local3375 auto dimIndex = bcastDimensions.getValues<int64_t>()[i]; in verify() local