Lines Matching refs:MatrixA
281 Value *MatrixA; in computeShapeInfoForInst() local
282 if (match(I, m_Store(m_Value(MatrixA), m_Value()))) { in computeShapeInfoForInst()
283 auto OpShape = ShapeMap.find(MatrixA); in computeShapeInfoForInst()
693 Value *MatrixA; in propagateShapeBackward() local
699 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeBackward()
701 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
702 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
708 m_Value(MatrixA), m_Value(M), m_Value(N)))) { in propagateShapeBackward()
710 if (setShapeInfo(MatrixA, {M, N})) in propagateShapeBackward()
711 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()
713 m_Value(MatrixA), m_Value(), m_Value(), m_Value(), in propagateShapeBackward()
715 if (setShapeInfo(MatrixA, {M, N})) { in propagateShapeBackward()
716 pushInstruction(MatrixA, WorkList); in propagateShapeBackward()