Lines Matching defs:ShapeInfo

184 struct ShapeInfo {
190   ShapeInfo(unsigned NumRows = 0, unsigned NumColumns = 0)
194   ShapeInfo(Value *NumRows, Value *NumColumns)
195       : ShapeInfo(cast<ConstantInt>(NumRows)->getZExtValue(),
198   bool operator==(const ShapeInfo &other) {
201   bool operator!=(const ShapeInfo &other) { return !(*this == other); }
223   ShapeInfo t() const { return ShapeInfo(NumColumns, NumRows); }
246 /// Return the ShapeInfo for the result of \p I, it it can be determined.
247 static std::optional<ShapeInfo>
249                         const DenseMap<Value *, ShapeInfo> &ShapeMap) {
255     return ShapeInfo(M, K);
259     return ShapeInfo(N, M);
264     return ShapeInfo(N, M);
267     return ShapeInfo(M, N);
492   DenseMap<Value *, ShapeInfo> ShapeMap;
544   MatrixTy getMatrix(Value *MatrixVal, const ShapeInfo &SI,
583   bool setShapeInfo(Value *V, ShapeInfo Shape) {
718         ShapeInfo Shape = ShapeMap[V];
739       Value *Op0, ShapeInfo Shape0, Value *Op1, ShapeInfo Shape1,
741       function_ref<Instruction *(Value *, ShapeInfo, Value *, ShapeInfo)>
829           [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) {
851           [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) {
871           [&](Value *T0, ShapeInfo Shape0, Value *T1, ShapeInfo Shape1) {
1152                       bool IsVolatile, ShapeInfo Shape, IRBuilder<> &Builder) {
1175                       ShapeInfo MatrixShape, Value *I, Value *J,
1176                       ShapeInfo ResultShape, Type *EltTy,
1193                  bool IsVolatile, ShapeInfo Shape) {
1217                    MaybeAlign MAlign, bool IsVolatile, ShapeInfo MatrixShape,
1255                   Value *Stride, bool IsVolatile, ShapeInfo Shape) {
1364     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1365     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1725     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1726     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1765   void createTiledLoops(CallInst *MatMul, Value *LPtr, ShapeInfo LShape,
1766                         Value *RPtr, ShapeInfo RShape, StoreInst *Store) {
1832     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1833     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1911       ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
1912       ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
1922         MA = getMatrix(A, ShapeInfo(R, M), Builder);
1923         MB = getMatrix(T, ShapeInfo(C, M), Builder);
1926         MA = getMatrix(T, ShapeInfo(R, M), Builder);
1927         MB = getMatrix(B, ShapeInfo(C, M), Builder);
2042     ShapeInfo LShape(MatMul->getArgOperand(2), MatMul->getArgOperand(3));
2043     ShapeInfo RShape(MatMul->getArgOperand(3), MatMul->getArgOperand(4));
2070     ShapeInfo ArgShape(Inst->getArgOperand(1), Inst->getArgOperand(2));
2136     ShapeInfo &Shape = I->second;
2186     ShapeInfo &Shape = I->second;