Lines Matching refs:MatrixA
566 Value *MatrixA; in propagateShapeForward() local
572 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeForward()
576 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeForward()
580 m_Value(MatrixA), m_Value(), m_Value(), in propagateShapeForward()
587 } else if (match(Inst, m_Store(m_Value(MatrixA), m_Value()))) { in propagateShapeForward()
588 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()
637 Value *MatrixA; in propagateShapeBackward() local
643 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeBackward()
645 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
646 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
652 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeBackward()
654 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
655 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
657 m_Value(MatrixA), m_Value(), m_Value(), m_Value(), in propagateShapeBackward()
659 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
660 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()