diff --git a/docs/.buildinfo b/docs/.buildinfo index 9081611fe..321c9f788 100644 --- a/docs/.buildinfo +++ b/docs/.buildinfo @@ -1,4 +1,4 @@ # Sphinx build info version 1 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done. -config: bfc625bef8624d43d40fb6e0a790344b +config: 0c26ac880e1785fd655d98bdaf85b41b tags: 645f666f9bcd5a90fca523b33c5a78b7 diff --git a/docs/_modules/brevitas/core/bit_width/const.html b/docs/_modules/brevitas/core/bit_width/const.html index 6739ff04b..54e0ef26e 100644 --- a/docs/_modules/brevitas/core/bit_width/const.html +++ b/docs/_modules/brevitas/core/bit_width/const.html @@ -1,11 +1,14 @@ + - + + + - brevitas.core.bit_width.const — Brevitas 0.9.0 documentation + brevitas.core.bit_width.const — Brevitas 0.10.0b1.dev37+g993c869.d20231123 documentation @@ -15,22 +18,23 @@ - - - + + + - - - - + + + + - - + + + @@ -43,79 +47,103 @@ - - + + - + + +
- - + - + + + + -
- -
+ - + Ctrl+K -
-
+
- - +
+
- -
-
+
- -
- -
- -
- -
- +
+ +
+
+
- -
- - -
-
-
+
- +
+
-
-
- -
- -
- - -
+
+
+
+
+ +
+ +
+ + + + +
+ +
+ + +
+
+ + + + + +
+

Source code for brevitas.core.bit_width.const

 # Copyright (C) 2023, Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: BSD-3-Clause
 
+from typing import Optional
+
 import torch
 from torch import Tensor
 from torch.nn import Module
@@ -364,7 +424,7 @@ 
Source code for brevitas.core.bit_width.const
[docs]class BitWidthConst(brevitas.jit.ScriptModule):
-    """
+    """
     ScriptModule that returns a constant bit-width wrapped in a float torch.tensor.
 
     Args:
@@ -383,10 +443,14 @@ 
Source code for brevitas.core.bit_width.const
        Maps to bit_width_impl_type == BitWidthImplType.CONST == 'CONST' == 'const' in higher-level APIs.
     """
 
-    def __init__(self, bit_width: int) -> None:
+    def __init__(
+            self,
+            bit_width: int,
+            dtype: Optional[torch.dtype] = None,
+            device: Optional[torch.device] = None) -> None:
         super(BitWidthConst, self).__init__()
         assert isinstance(bit_width, int)
-        self.bit_width = StatelessBuffer(torch.tensor(float(bit_width)))
+        self.bit_width = StatelessBuffer(torch.tensor(float(bit_width), dtype=dtype, device=device))
 
 
[docs]    @brevitas.jit.script_method
     def forward(self) -> Tensor:
@@ -394,7 +458,7 @@ 
Source code for brevitas.core.bit_width.const
[docs]class BitWidthStatefulConst(brevitas.jit.ScriptModule):
-    """
+    """
     ScriptModule that returns a constant bit-width wrapped in a float torch.tensor but retains the
     bit-width as part of the module state.
 
@@ -416,10 +480,15 @@ 
Source code for brevitas.core.bit_width.const
        'stateful_const' in higher-level APIs.
     """
 
-    def __init__(self, bit_width: int) -> None:
+    def __init__(
+            self,
+            bit_width: int,
+            dtype: Optional[torch.dtype] = None,
+            device: Optional[torch.device] = None) -> None:
         super(BitWidthStatefulConst, self).__init__()
         assert isinstance(bit_width, int)
-        self.register_buffer("bit_width", torch.tensor(float(bit_width)))
+        self.register_buffer(
+            "bit_width", torch.tensor(float(bit_width), dtype=dtype, device=device))
 
 
[docs]    @brevitas.jit.script_method
     def forward(self) -> Tensor:
@@ -463,86 +532,71 @@ 
Source code for brevitas.core.bit_width.const
return output_bit_width
-
- - - -
- -
-
-
- -
- - - -
+
-
- - -
- -
- -
- -
- + + + + +
+ +
- +
+
- - -
-
-
+ + +
-
+
+ +
- - - - + + - + +
+ -
- - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
+
- + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
+
- + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
+
- + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
+
- + - - - - + + - + + + - - - - -
-
- - -
+
-
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + - - - - + + - + - - - - + + - + + + - - - - -
-
- - -
- -
- -
+
-
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+
- -
- -
- - -
- -
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + + + - - - - -
-
- -
+
+ +
+
On this page -
-
+ -
- -
- - -
- -
- -
+
-
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + - - - - + + - + + + - - - - -
-
- - -
+
-
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + - - - - + + - + + + - - - - -
-
- - -
+
-
- -
- -
- -
+
+
- Show Source + Show Source +
-
- -
- - -
-
-
+
+
- + - - - - + + - + - - - - + + - + - - - - + + - + - - - - + + - + - - - - + + - + - - - - + + - + - - - - + + - + - - - - + + -