add fatten/flatten2 trt converter test cases

Author: baoachun

python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_flatten.py

@@ -21,6 +21,9 @@
 
 
 class TrtConvertFlattenTest_dim_2(TrtLayerAutoScanTest):
+    def is_program_valid(self, program_config: ProgramConfig) -> bool:
+        return True
+
     def sample_program_configs(self):
         def generate_input(batch):
             return np.random.random([batch, 32]).astype(np.float32)
@@ -102,6 +105,9 @@ def test(self):
 
 
 class TrtConvertFlattenTest_dim_3(TrtLayerAutoScanTest):
+    def is_program_valid(self, program_config: ProgramConfig) -> bool:
+        return True
+
     def sample_program_configs(self):
         def generate_input(batch):
             return np.random.random([batch, 32, 64]).astype(np.float32)
@@ -183,6 +189,9 @@ def test(self):
 
 
 class TrtConvertFlattenTest_dim_4(TrtLayerAutoScanTest):
+    def is_program_valid(self, program_config: ProgramConfig) -> bool:
+        return True
+
     def sample_program_configs(self):
         def generate_input(batch):
             return np.random.random([batch, 8, 8, 8]).astype(np.float32)
@@ -264,6 +273,9 @@ def test(self):
 
 
 class TrtConvertFlattenTest_dim_5(TrtLayerAutoScanTest):
+    def is_program_valid(self, program_config: ProgramConfig) -> bool:
+        return True
+
     def sample_program_configs(self):
         def generate_input(batch):
             return np.random.random([batch, 8, 8, 8]).astype(np.float32)