Skip to content
Snippets Groups Projects
Commit 54de6a23 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[HLSCustomOp] update expected cycles for StreamingMaxPool_Batch

based on loop nest
parent f8bbb4a5
No related branches found
No related tags found
No related merge requests found
......@@ -96,8 +96,10 @@ class StreamingMaxPool_Batch(HLSCustomOp):
return np.prod(folded_oshape[:-1])
def get_exp_cycles(self):
# Channels * batch size * odim * odim
return np.prod(self.get_folded_output_shape()[:-1])
# derived from StreamingMaxPool_Batch loop nest
k = self.get_nodeattr("PoolDim")
ifm_dim = self.get_nodeattr("ImgDim")
return ifm_dim * (ifm_dim + (ifm_dim / k))
def get_instream_width(self):
dt_bits = self.get_input_datatype().bitwidth()
......
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