Lines Matching defs:broadcastedDims
2376 /// `broadcastedDims` dimensions in the dstShape are broadcasted.
2381 /// `value`, `dstShape` and `broadcastedDims` must be properly specified or
2384 /// 2. `broadcastedDims` must be confined to [0 .. rank(value.getVectorType)]
2385 /// 2. `dstShape` trimmed of the dimensions specified in `broadcastedDims`
2389 const llvm::SetVector<int64_t> &broadcastedDims) {
2395 if (broadcastedDims.contains(i))
2399 assert(broadcastedDims.size() == dstShape.size() - checkShape.size() &&
2400 "ill-formed broadcastedDims contains values not confined to "
2431 // broadcastedDims = [0, 2, 4]
2443 int64_t nextSrcShapeDim = broadcastedDims.size();
2445 if (broadcastedDims.contains(i)) {