Searched refs:MatrixB (Results 1 – 1 of 1) sorted by relevance
567 Value *MatrixB; in propagateShapeForward() local572 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeForward()638 Value *MatrixB; in propagateShapeBackward() local643 m_Value(MatrixA), m_Value(MatrixB), m_Value(M), in propagateShapeBackward()648 if (setShapeInfo(MatrixB, {N, K})) in propagateShapeBackward()649 pushInstruction(MatrixB, WorkList); in propagateShapeBackward()