|
|
@ -49,7 +49,7 @@ class TestReorgOp(OpTest):
|
|
|
|
self.x.shape[1], self.x.shape[0], self.stride, self.forward,
|
|
|
|
self.x.shape[1], self.x.shape[0], self.stride, self.forward,
|
|
|
|
self.out_1d)
|
|
|
|
self.out_1d)
|
|
|
|
self.out = np.reshape(self.out_1d, self.infered_shape)
|
|
|
|
self.out = np.reshape(self.out_1d, self.infered_shape)
|
|
|
|
self.attrs = {"stride": long(self.stride)}
|
|
|
|
self.attrs = {"stride": self.stride}
|
|
|
|
self.outputs = {"Out": self.out}
|
|
|
|
self.outputs = {"Out": self.out}
|
|
|
|
|
|
|
|
|
|
|
|
def init_data(self):
|
|
|
|
def init_data(self):
|
|
|
|