From 8ad2d942e787469ab4fbbf5438ef21a09dcfe0be Mon Sep 17 00:00:00 2001 From: Asthestarsfalll <1186454801@qq.com> Date: Mon, 13 May 2024 11:56:45 +0800 Subject: [PATCH] fix generate script --- .../tests/test_models/gen_scripts/generate_tril_triu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/frontends/paddle/tests/test_models/gen_scripts/generate_tril_triu.py b/src/frontends/paddle/tests/test_models/gen_scripts/generate_tril_triu.py index cfc5d087eaa54b..14fd9e11e4ecd6 100644 --- a/src/frontends/paddle/tests/test_models/gen_scripts/generate_tril_triu.py +++ b/src/frontends/paddle/tests/test_models/gen_scripts/generate_tril_triu.py @@ -29,7 +29,7 @@ def triu(name: str, x, diagonal=0, dtype="float32"): saveModel( name, exe, - feedkeys=["x"], + feed_vars=[node_x], fetchlist=[triu_outs], inputs=[x], outputs=outs, @@ -54,7 +54,7 @@ def tril(name: str, x, diagonal=0, dtype="float32"): saveModel( name, exe, - feedkeys=["x"], + feed_vars=[node_x], fetchlist=[tril_outs], inputs=[x], outputs=outs,