Lines Matching defs:broadcast

1259   // Fold extractelement(broadcast(X)) -> X.
1260   if (auto broadcast = getVector().getDefiningOp<vector::BroadcastOp>())
1261     if (!llvm::isa<VectorType>(broadcast.getSource().getType()))
1262       return broadcast.getSource();
1676   // If splat or broadcast from a scalar, just return the source scalar.
2057     // Special case if broadcast src is a 0D vector.
2367   // Scalar broadcast is without any unit dim broadcast.
2377 /// This requires (and asserts) that the broadcast is free of dim-1
2379 /// Since vector.broadcast only allows expanding leading dimensions, an extra
2380 /// vector.transpose may be inserted to make the broadcast possible.
2408   // Step 2. If scalar -> dstShape broadcast, just do it.
2418   // Step 3. Since vector.broadcast only allows creating leading dims,
2419   //   vector -> dstShape broadcast may require a transpose.
2422   //      achievable by a simple broadcast.
2437   //            leading broadcast part      src shape part
2466          "unexpected dim-1 broadcast");
2489   // From now on, only vectors broadcast.
2502     // Have mismatching dims (in the sense of vector.broadcast semantics) been
2556   llvm_unreachable("unexpected vector.broadcast op error");
2678   // but must be a canonicalization into a vector.broadcast.
2721 // to a broadcast.
2940 // broadcast.
3839     auto broadcast = op.getVector().getDefiningOp<BroadcastOp>();
3840     if (!broadcast)
3843         llvm::dyn_cast<VectorType>(broadcast.getSource().getType());
3848     // Check if the most inner dimensions of the source of the broadcast are the
3850     // use a broadcast as the original dimensions are untouched.
3858     Value source = broadcast.getSource();
3860     // source of the orignal broadcast and then broadcast the extracted value.
4338   // 1. Process non-broadcast dims
4360   // 2. Handle broadcast dims
4361   // If all non-broadcast dims are "in bounds", then all bcast dims should be
4456 /// can replace the transfer_read + transfer_write by vector.broadcast and
4471 /// %0 = vector.broadcast %arg1 : vector<4x1xf32> to vector<100x5x4x1xf32>
4516     // Calculate the broadcast shape by applying the reverse permutation to the
4684   // We do not allow broadcast dimensions on TransferWriteOps for the moment,
4687     return emitOpError("should not have broadcast dimensions");
5572   // Cancelling broadcast and shape cast ops.
5712 /// This only applies when the shape of the broadcast source
5713 /// 1. is a suffix of the shape of the result (i.e. when broadcast without
5734     // with a broadcast to the final shape.
6014 // Folds transpose(broadcast(<scalar>)) into brodcast(<scalar>).