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"])