Searched refs:taus_shape (Results 1 – 1 of 1) sorted by relevance
5962 auto taus_shape = llvm::to_vector<4>(batch_dims); in QRBlock() local5963 taus_shape.push_back(n); in QRBlock()5965 loc, RankedTensorType::get(taus_shape, a_type.getElementType()), zero, in QRBlock()5966 GetI64ElementsAttr(taus_shape, rewriter)); in QRBlock()