Skip to content

Commit

Permalink
Update test_converter_activation.py
Browse files Browse the repository at this point in the history
  • Loading branch information
PolaKuma committed Jan 3, 2025
1 parent 00603b9 commit 2728c5f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions test/tensorrt/test_converter_activation.py
Original file line number Diff line number Diff line change
Expand Up @@ -210,6 +210,7 @@ def setUp(self):
self.api_args = {
"x": np.random.randn(2, 3).astype("float32"),
}
self.program_config = {"feed_list": ["x"]}
self.min_shape = {"x": [1, 3]}
self.max_shape = {"x": [5, 3]}

Expand Down

0 comments on commit 2728c5f

Please sign in to comment.