@@ -41,7 +41,7 @@ ______________________________________________________________________
4141### <kbd >method</kbd > ` compute_op_predecessors `
4242
4343``` python
44- compute_op_predecessors() → DefaultDict[Optional [QuantizedOp], List[Tuple[Optional [QuantizedOp], str ]]]
44+ compute_op_predecessors() → DefaultDict[Union [QuantizedOp, NoneType ], List[Tuple[Union [QuantizedOp, NoneType ], str ]]]
4545```
4646
4747Compute the predecessors for each QuantizedOp in a QuantizedModule.
@@ -61,7 +61,7 @@ ______________________________________________________________________
6161``` python
6262detect_patterns(
6363 predecessors: DefaultDict[Optional[QuantizedOp], List[Tuple[Optional[QuantizedOp], str ]]]
64- ) → Dict[QuantizedMixingOp, Tuple[List[Optional [QuantizedOp]], Optional [QuantizedOp]]]
64+ ) → Dict[QuantizedMixingOp, Tuple[List[Union [QuantizedOp, NoneType ]], Union [QuantizedOp, NoneType ]]]
6565```
6666
6767Detect the patterns that can be optimized with roundPBS in the QuantizedModule.
@@ -107,7 +107,7 @@ ______________________________________________________________________
107107### <kbd >method</kbd > ` process `
108108
109109``` python
110- process() → Dict[QuantizedMixingOp, Tuple[List[Optional [QuantizedOp]], Optional [QuantizedOp]]]
110+ process() → Dict[QuantizedMixingOp, Tuple[List[Union [QuantizedOp, NoneType ]], Union [QuantizedOp, NoneType ]]]
111111```
112112
113113Analyze an ONNX graph and detect Gemm/Conv patterns that can use RoundPBS.
@@ -129,7 +129,7 @@ ______________________________________________________________________
129129``` python
130130process_patterns(
131131 valid_paths: Dict[QuantizedMixingOp, Tuple[List[Optional[QuantizedOp]], Optional[QuantizedOp]]]
132- ) → Dict[QuantizedMixingOp, Tuple[List[Optional [QuantizedOp]], Optional [QuantizedOp]]]
132+ ) → Dict[QuantizedMixingOp, Tuple[List[Union [QuantizedOp, NoneType ]], Union [QuantizedOp, NoneType ]]]
133133```
134134
135135Configure the rounding bits of roundPBS for the optimizable operations.
0 commit comments