Searched refs:lhsShape (Results 1 – 2 of 2) sorted by relevance
325 auto lhsShape = llvm::to_vector<6>( in fold() local327 if (lhsShape.empty()) in fold()333 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()346 auto lhsShape = llvm::to_vector<6>( in fold() local351 resultShape.append(lhsShape.begin(), lhsShape.end()); in fold()448 auto lhsShape = llvm::to_vector<6>( in fold() local453 if (OpTrait::util::staticallyKnownBroadcastable(lhsShape, rhsShape)) in fold()459 SmallVector<int64_t, 6> lhsShape, rhsShape; in fold() local460 if (failed(getShapeVec(lhs(), lhsShape))) in fold()465 if (OpTrait::util::staticallyKnownBroadcastable(lhsShape, rhsShape)) in fold()
473 auto lhsShape = getLhsType().getShape(); in getIterationBounds() local485 iterationBounds.push_back(lhsShape[lhsDimIndex]); in getIterationBounds()