diff --git a/python/tvm/relay/frontend/paddlepaddle.py b/python/tvm/relay/frontend/paddlepaddle.py index 7042154709ae..a869e2e1b807 100644 --- a/python/tvm/relay/frontend/paddlepaddle.py +++ b/python/tvm/relay/frontend/paddlepaddle.py @@ -2268,7 +2268,7 @@ def from_translated_layer(self, layer, shape_dict): self.shape_dict = shape_dict program = layer.program() parameters = dict() - for param in layer.parameters(): + for param in layer.parameters() + layer.buffers(): parameters[param.name] = np.array(param.value().get_tensor()) self.check_unsupported_ops(program) self.extract_parameters(program, parameters)