Skip to content
Snippets Groups Projects
Commit 8c32a78f authored by auphelia's avatar auphelia
Browse files

[CustomOp] Change calculation of exp cycles for GlobalAccPool

parent ff4499eb
No related branches found
No related tags found
No related merge requests found
......@@ -183,8 +183,8 @@ class GlobalAccPool_Batch(HLSCustomOp):
return np.prod(self.get_folded_output_shape()[1:-1])
def get_exp_cycles(self):
# Channels/PE * batch size (ofmdim = 1)
return np.prod(self.get_folded_output_shape()[:-1])
# Channels/PE * batch size * idim * idim
return np.prod(self.get_folded_input_shape()[:-1])
def execute_node(self, context, graph):
mode = self.get_nodeattr("exec_mode")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment