Skip to content

Commit

Permalink
chore: prepare release 1.5.0
Browse files Browse the repository at this point in the history
  • Loading branch information
RomanBredehoft committed Apr 5, 2024
1 parent 8d1d98a commit 1ea5d46
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ ______________________________________________________________________
### <kbd>method</kbd> `bitwidth_and_range_report`

```python
bitwidth_and_range_report() → Optional[Dict[str, Dict[str, Union[Tuple[int, ], int]]]]
bitwidth_and_range_report() → Union[Dict[str, Dict[str, Union[Tuple[int, ], int]]], NoneType]
```

Report the ranges and bit-widths for layers that mix encrypted integer values.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ ______________________________________________________________________
### <kbd>method</kbd> `compute_op_predecessors`

```python
compute_op_predecessors() → DefaultDict[Optional[QuantizedOp], List[Tuple[Optional[QuantizedOp], str]]]
compute_op_predecessors() → DefaultDict[Union[QuantizedOp, NoneType], List[Tuple[Union[QuantizedOp, NoneType], str]]]
```

Compute the predecessors for each QuantizedOp in a QuantizedModule.
Expand All @@ -61,7 +61,7 @@ ______________________________________________________________________
```python
detect_patterns(
predecessors: DefaultDict[Optional[QuantizedOp], List[Tuple[Optional[QuantizedOp], str]]]
) → Dict[QuantizedMixingOp, Tuple[List[Optional[QuantizedOp]], Optional[QuantizedOp]]]
) → Dict[QuantizedMixingOp, Tuple[List[Union[QuantizedOp, NoneType]], Union[QuantizedOp, NoneType]]]
```

Detect the patterns that can be optimized with roundPBS in the QuantizedModule.
Expand Down Expand Up @@ -107,7 +107,7 @@ ______________________________________________________________________
### <kbd>method</kbd> `process`

```python
process() → Dict[QuantizedMixingOp, Tuple[List[Optional[QuantizedOp]], Optional[QuantizedOp]]]
process() → Dict[QuantizedMixingOp, Tuple[List[Union[QuantizedOp, NoneType]], Union[QuantizedOp, NoneType]]]
```

Analyze an ONNX graph and detect Gemm/Conv patterns that can use RoundPBS.
Expand All @@ -129,7 +129,7 @@ ______________________________________________________________________
```python
process_patterns(
valid_paths: Dict[QuantizedMixingOp, Tuple[List[Optional[QuantizedOp]], Optional[QuantizedOp]]]
) → Dict[QuantizedMixingOp, Tuple[List[Optional[QuantizedOp]], Optional[QuantizedOp]]]
) → Dict[QuantizedMixingOp, Tuple[List[Union[QuantizedOp, NoneType]], Union[QuantizedOp, NoneType]]]
```

Configure the rounding bits of roundPBS for the optimizable operations.
Expand Down

0 comments on commit 1ea5d46

Please sign in to comment.