Skip to content

Commit ce715c5

Browse files
Fix docstring of to_backend in backend_api.py (#8879)
The second method overloading of to_backend() has the wrong signature in docstring Signed-off-by: Yufeng Shi <[email protected]>
1 parent 542480c commit ce715c5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

exir/backend/backend_api.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
# Copyright (c) Meta Platforms, Inc. and affiliates.
22
# All rights reserved.
3+
# Copyright 2025 Arm Limited and/or its affiliates.
34
#
45
# This source code is licensed under the BSD-style license found in the
56
# LICENSE file in the root directory of this source tree.
@@ -56,9 +57,9 @@ def to_backend(
5657
) -> LoweredBackendModule:
5758
5859
def to_backend(
59-
graph_module: torch.fx.GraphModule,
60-
partitioner: Type[TPartitioner],
61-
) -> torch.fx.GraphModule
60+
edge_program: ExportedProgram,
61+
partitioner: Partitioner,
62+
) -> ExportedProgram:
6263
"""
6364
pass
6465

0 commit comments

Comments
 (0)