22fb2fe9ea4ad5ef536709aa07838d3d6c7357ee,nn_dataflow/LoopBlockingScheme.py,LoopBlockingScheme,_set_unit_cnt,#LoopBlockingScheme#,336

Before Change

            blp1 = bl + 1
            pblti = Util.prod(self.ti[blp1:])
            pblto = Util.prod(self.to[blp1:])
            pbltb = Util.prod(self.tb[blp1:])

            uc = [1] * de.NUM
            uc[de.FIL] = pblti * pblto
            uc[de.IFM] = pblti * pbltb
            uc[de.OFM] = pblto * pbltb

            self.unit_cnt.append(uc)

    def _set_fetch(self):

After Change

        for bl in range(self.BL.NUM):
            // BL corresponds to the BL + 1 element in ti/to/tb.
            blp1 = bl + 1
            bl_tps = [Util.prod(ts) for ts in self._bl_t(slice(blp1, None))]
            uc = self._t_data_cnt(bl_tps)
            self.unit_cnt.append(uc)

Instances