From 1ea5d4654a33816ff64e76193131bb5f12de0854 Mon Sep 17 00:00:00 2001 From: RomanBredehoft <56846628+RomanBredehoft@users.noreply.github.com> Date: Fri, 5 Apr 2024 16:04:47 +0000 Subject: [PATCH] chore: prepare release 1.5.0 --- .../api/concrete.ml.quantization.quantized_module.md | 2 +- .../concrete.ml.quantization.quantized_module_passes.md | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docs/references/api/concrete.ml.quantization.quantized_module.md b/docs/references/api/concrete.ml.quantization.quantized_module.md index fcf52f1dd..6dd778aa1 100644 --- a/docs/references/api/concrete.ml.quantization.quantized_module.md +++ b/docs/references/api/concrete.ml.quantization.quantized_module.md @@ -72,7 +72,7 @@ ______________________________________________________________________ ### method `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. diff --git a/docs/references/api/concrete.ml.quantization.quantized_module_passes.md b/docs/references/api/concrete.ml.quantization.quantized_module_passes.md index 56b77818a..205e213a6 100644 --- a/docs/references/api/concrete.ml.quantization.quantized_module_passes.md +++ b/docs/references/api/concrete.ml.quantization.quantized_module_passes.md @@ -41,7 +41,7 @@ ______________________________________________________________________ ### method `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. @@ -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. @@ -107,7 +107,7 @@ ______________________________________________________________________ ### method `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. @@ -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.