Lines Matching refs:MatrixA
614 Value *MatrixA; in propagateShapeForward() local
620 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeForward()
624 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeForward()
628 m_Value(MatrixA), m_Value(), m_Value(), in propagateShapeForward()
635 } else if (match(Inst, m_Store(m_Value(MatrixA), m_Value()))) { in propagateShapeForward()
636 auto OpShape = ShapeMap.find(MatrixA); in propagateShapeForward()
685 Value *MatrixA; in propagateShapeBackward() local
691 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeBackward()
693 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
694 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
700 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeBackward()
702 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
703 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
705 m_Value(MatrixA), m_Value(), m_Value(), m_Value(), in propagateShapeBackward()
707 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
708 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()