4fb88fdc63292d93ce2fdbbf77a4043f18b2c769,nn_dataflow/PartitionScheme.py,PartitionScheme,part_layer,#PartitionScheme#Any#Any#,98

Before Change

        """
        p_layer = Layer(layer.nifm,
                        Util.idivc(layer.nofm, self.pdims[pe.OUTP].size()),
                        (Util.idivc(layer.hofm, self.pdims[pe.OFMP].h),
                         Util.idivc(layer.wofm, self.pdims[pe.OFMP].w)),
                        layer.sfil,
                        layer.strd)

After Change

        p_hofm = Util.idivc(layer.hofm, self.pdims[pe.OFMP].h)
        p_wofm = Util.idivc(layer.wofm, self.pdims[pe.OFMP].w)

        if isinstance(layer, ConvLayer):
            p_layer = ConvLayer(layer.nifm, p_nofm, (p_hofm, p_wofm),
                                layer.sfil,
                                strd=(layer.htrd, layer.wtrd))
        elif isinstance(layer, LocalRegionLayer):
            p_layer = LocalRegionLayer(p_nofm, (p_hofm, p_wofm),
                                       layer.nreg, (layer.hreg, layer.wreg),
                                       strd=(layer.htrd, layer.wtrd))
        else:
            raise TypeError("PartitionScheme: unrecognized layer type.")

        p_batch_size = batch_size / self.pdims[pe.BATP].size()

        p_occ = 1. * layer.total_ops(batch_size) \
                / (p_layer.total_ops(p_batch_size) * self.size())

Instances