Lines Matching defs:indexPool

661                          IndexPool &indexPool, int64_t index) {
662   auto [it, inserted] = indexPool.try_emplace(index);
670                           IndexPool &indexPool, Value tensor, int64_t index) {
671   auto indexValue = createIndex(rewriter, loc, indexPool, index);
676                                        IndexPool &indexPool, Value tensor,
682     return getTensorDim(rewriter, loc, indexPool, tensor, index);
705 computeTargetSize(PatternRewriter &rewriter, Location loc, IndexPool &indexPool,
730       getTensorDim(rewriter, loc, indexPool, operandsWithDynamicDim[0], dim);
737         getTensorDim(rewriter, loc, indexPool, operandsWithDynamicDim[i], dim);
747                    IndexPool &indexPool, ValueRange operands) {
754         computeTargetSize(rewriter, loc, indexPool, operands, dim);
762                                        IndexPool &indexPool, Value operand,
791   auto one = createIndex(rewriter, loc, indexPool, 1);
792   auto runtimeSize = getTensorDim(rewriter, loc, indexPool, operand, dim);
845                                         IndexPool &indexPool, Value operand,
853         broadcastDynamicDimension(rewriter, loc, indexPool, operand, index,
860                            IndexPool &indexPool, ValueRange operands,
869     return broadcastDynamicDimensions(rewriter, loc, indexPool, operand,
950   IndexPool indexPool;
962       computeTargetShape(rewriter, loc, indexPool, expandedOperands);
964       rewriter, loc, indexPool, expandedOperands, targetShape, masterOperands);