No public description

PiperOrigin-RevId: 657246996 Change-Id: I93bdf156d77db53d393284383eb4b9aeea35a087
google · Jul 29, 2024 · dec22c7 · dec22c7
1 parent 6be2f5c
commit dec22c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qkeras/qtools/DnC/divide_and_conquer.py b/qkeras/qtools/DnC/divide_and_conquer.py
@@ -274,7 +274,7 @@ def get_per_layer_cost(layer_quantizer_bitwidth, layer_mac_count, layer_shapes,
       dnc_layer_cost_ace.MemoryGatesPerBit["Register"] +
       OutElementPerClk * layer_quantizer_bitwidth["output_bits"] *
       dnc_layer_cost_ace.MemoryGatesPerBit["Register"] +
-      np.product(layer_shapes["weight_shape"]) *
+      np.prod(layer_shapes["weight_shape"]) *
       layer_quantizer_bitwidth["weight_bits"] *
       dnc_layer_cost_ace.MemoryGatesPerBit["ROM"])