No public description

PiperOrigin-RevId: 579349096 Change-Id: Ieff86598351a0d9a614299fe77d8b0db2f2559ed
google · Nov 4, 2023 · c5051b5 · c5051b5
1 parent f55d93f
commit c5051b5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/qkeras/qtools/divide_and_conquer.py → qkeras/qtools/DnC/divide_and_conquer.py b/qkeras/qtools/divide_and_conquer.py → qkeras/qtools/DnC/divide_and_conquer.py
@@ -36,7 +36,7 @@
 from qkeras.qtools import qgraph
 from qkeras.qtools import qtools_util
 from qkeras.qtools import generate_layer_data_type_map
-from qkeras.qtools import dnc_layer_cost_ace
+from qkeras.qtools.DnC import dnc_layer_cost_ace
 
 
 class CostMode(enum.Enum):
@@ -262,10 +262,13 @@ def get_per_layer_cost(layer_quantizer_bitwidth, layer_mac_count, layer_shapes,
       wbit=layer_quantizer_bitwidth["weight_bits"],
       abit=layer_quantizer_bitwidth["acc_bits"],
       regen_params=False)
-  pe_area = (mac_gates * layer_mac_count * cin_unroll * cout_unroll *
-             kh_unroll * kw_unroll)
+
+  # pe_area is not dependent on total num of MACs in the layer.
+  pe_area = (mac_gates * cin_unroll * cout_unroll * kh_unroll * kw_unroll)
 
   # Memory includes input, output and weight memory, translated to gates.
+  # TODO(lishanok@): weights could be stored in either SRAM or ROM, dependent
+  # on user specification.
   memory_area = (
       InElementPerClk * layer_quantizer_bitwidth["input_bits"] *
       dnc_layer_cost_ace.MemoryGatesPerBit["Register"] +

diff --git a/qkeras/qtools/dnc_layer_cost_ace.py → qkeras/qtools/DnC/dnc_layer_cost_ace.py b/qkeras/qtools/dnc_layer_cost_ace.py → qkeras/qtools/DnC/dnc_layer_cost_ace.py
diff --git a/tests/qtools_model_test.py b/tests/qtools_model_test.py
@@ -38,7 +38,7 @@
 from qkeras.utils import model_save_quantized_weights
 from qkeras.qtools.quantized_operators import adder_impl
 from qkeras.qtools.quantized_operators import quantizer_impl
-from qkeras.qtools import divide_and_conquer
+from qkeras.qtools.DnC import divide_and_conquer
 
 
 def qdense_model_fork():