diff --git a/python/paddle/fluid/tests/unittests/test_attribute_var.py b/python/paddle/fluid/tests/unittests/test_attribute_var.py index a79a8d400a36339bd6c58279d5def541bea648dd..950d33a9bbc1e76754100191e110622c47eef608 100644 --- a/python/paddle/fluid/tests/unittests/test_attribute_var.py +++ b/python/paddle/fluid/tests/unittests/test_attribute_var.py @@ -93,7 +93,7 @@ class TestTileTensorList(UnittestBase): self.shapes = [[2, 3, 4]] self.save_path = os.path.join(self.temp_dir.name, 'tile_tensors') - def test_static(self): + def _test_static(self): main_prog = Program() starup_prog = Program() with program_guard(main_prog, starup_prog): @@ -127,7 +127,7 @@ class TestTileTensor(UnittestBase): self.shapes = [[2, 3, 4]] self.save_path = os.path.join(self.temp_dir.name, 'tile_tensor') - def test_static(self): + def _test_static(self): main_prog = Program() starup_prog = Program() with program_guard(main_prog, starup_prog): diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py index 908e117ed8ab4eb2188ddd7b7c87a34e775df20d..cd2e303fd9cba5fca1eda4162cc73a5113f459a7 100755 --- a/python/paddle/tensor/manipulation.py +++ b/python/paddle/tensor/manipulation.py @@ -1117,7 +1117,9 @@ def concat(x, axis=0, name=None): attrs = {} if isinstance(axis, Variable): axis.stop_gradient = True - attrs['axis'] = axis + inputs['AxisTensor'] = axis + else: + attrs['axis'] = axis helper.append_op(type='concat', inputs=inputs, @@ -2935,11 +2937,13 @@ def tile(x, repeat_times, name=None): if isinstance(repeat_times, Variable): repeat_times.stop_gradient = True - attrs['repeat_times'] = repeat_times + inputs['RepeatTimes'] = repeat_times + attrs['repeat_times'] = [-1] elif isinstance(repeat_times, (list, tuple)): attrs['repeat_times'] = get_attr_repeat_times(repeat_times) if utils._contain_var(repeat_times): - attrs['repeat_times'] = utils._convert_to_tensor_list(repeat_times) + inputs['repeat_times_tensor'] = utils._convert_to_tensor_list( + repeat_times) dtype = helper.input_dtype(input_param_name='x') out = helper.create_variable_for_type_inference(dtype)