Lines Matching defs:broadcastShape

2421   //   1. The leading entries of the broadcastShape that is guaranteed to be
2424   //      bring us from `broadcastShape` back to he desired `dstShape`.
2426   SmallVector<int64_t> broadcastShape, permutation(dstShape.size(), -1);
2427   broadcastShape.reserve(dstShape.size());
2434   //   broadcastShape  = 1x3x5x2x4
2439   // Note that the trailing dims of broadcastShape are exactly the srcShape
2447       // bring it to the head of the broadcastShape.
2448       // It will need to be permuted back from `broadcastShape.size() - 1` into
2450       broadcastShape.push_back(dstShape[i]);
2451       permutation[i] = broadcastShape.size() - 1;
2455       // Don't touch `broadcastShape` here, the whole srcShape will be
2461   llvm::append_range(broadcastShape, srcVectorType.getShape());
2464   assert(::computeBroadcastedUnitDims(srcVectorType.getShape(), broadcastShape)
2468   VectorType broadcastType = VectorType::get(broadcastShape, elementType);
4519     SmallVector<int64_t> broadcastShape(destShape.size());
4522       broadcastShape[pos.value()] = destShape[pos.index()];
4527         broadcastShape, defWrite.getVectorType().getElementType(),