Lines Matching defs:arrayShape

765   llvm::ArrayRef<int64_t> arrayShape = arrayTy.getShape();
776       if (maskShape.size() != arrayShape.size())
782         for (std::size_t i = 0; i < arrayShape.size(); ++i) {
783           int64_t arrayExtent = arrayShape[i];
812   llvm::ArrayRef<int64_t> arrayShape = arrayTy.getShape();
823     if (arrayShape.size() > 1 && dim != nullptr) {
833       if (resultShape.size() != (arrayShape.size() - 1))
881   llvm::ArrayRef<int64_t> arrayShape = arrayTy.getShape();
893   if (arrayShape.size() > 1 && dim != nullptr) {
898     if (resultShape.size() != (arrayShape.size() - 1))
970   llvm::ArrayRef<int64_t> arrayShape = arrayTy.getShape();
973   if (dim && arrayShape.size() == 1) {
986     if (dim && resultShape.size() != (arrayShape.size() - 1))