fixed concat tests

Author: jakpiase

python/paddle/fluid/tests/unittests/mkldnn/test_concat_bf16_mkldnn_op.py

@@ -40,11 +40,14 @@ def setUp(self):
  'mkldnn_data_type': self.mkldnn_data_type
  }
 
+ self.sections = [self.x0.shape[self.axis]] * 2
+ self.sections[1] += self.x1.shape[self.axis]
+
  self.output = np.concatenate(
  (self.x0, self.x1, self.x2), axis=self.axis).astype(np.uint16)
  self.outputs = {'Out': self.output}
 
- def calculate_grads(self): 
+ def calculate_grads(self):
  self.dout = self.outputs['Out']
  self.dxs = np.split(self.dout, self.sections, self.axis)
 
@@ -73,9 +76,9 @@ def init_axis(self):
  self.axis = 0
 
  def init_shape(self):
- self.x0_shape = [2, 2, 1, 2]
- self.x1_shape = [1, 2, 1, 2]
- self.x2_shape = [3, 2, 1, 2]
+ self.x0_shape = [6, 2, 4, 3]
+ self.x1_shape = [7, 2, 4, 3]
+ self.x2_shape = [8, 2, 4, 3]
 
 
 # --------------------test concat bf16 in with axis 1--------------------
@@ -86,9 +89,9 @@ def init_axis(self):
  self.axis = 1
 
  def init_shape(self):
- self.x0_shape = [1, 1, 5, 5]
- self.x1_shape = [1, 2, 5, 5]
- self.x2_shape = [1, 3, 5, 5]
+ self.x0_shape = [1, 4, 5, 5]
+ self.x1_shape = [1, 8, 5, 5]
+ self.x2_shape = [1, 6, 5, 5]
 
 
 # --------------------test concat bf16 in with axis 2--------------------

python/paddle/fluid/tests/unittests/mkldnn/test_concat_mkldnn_op.py

@@ -41,9 +41,6 @@ def setUp(self):
  self.output = np.concatenate(
  (self.x0, self.x1, self.x2), axis=self.axis).astype(self.dtype)
 
- self.sections = [self.x0.shape[self.axis]] * 2
- self.sections[1] += self.x1.shape[self.axis]
-
  self.outputs = {'Out': self.output}
 
  def configure_datatype(self):