48084d8851f801b5b2f550708cecf67b63eb92a8,src/sdk/pynni/tests/test_compressor.py,,,#,58

Before Change

    return test_tf2_func


k1 = [[1] * 3] * 3
k2 = [[2] * 3] * 3
k3 = [[3] * 3] * 3
k4 = [[4] * 3] * 3
k5 = [[5] * 3] * 3

w = [[k1, k2, k3, k4, k5]] * 10


class CompressorTestCase(TestCase):
    def test_torch_level_pruner(self):

After Change

    return test_tf2_func

// for fpgm filter pruner test
w = np.array([[[[i+1]*3]*3]*5 for i in range(10)])


class CompressorTestCase(TestCase):

Instances