Lines Matching refs:MatrixA
565 Value *MatrixA; in propagateShapeForward() local
571 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeForward()
575 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeForward()
579 m_Value(MatrixA), m_Value(), m_Value(), in propagateShapeForward()
586 } else if (match(Inst, m_Store(m_Value(MatrixA), m_Value()))) { in propagateShapeForward()
587 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()
636 Value *MatrixA; in propagateShapeBackward() local
642 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeBackward()
644 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
645 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
651 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeBackward()
653 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
654 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
656 m_Value(MatrixA), m_Value(), m_Value(), m_Value(), in propagateShapeBackward()
658 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
659 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()