Skip to content

Commit 7db5338

Browse files
committed
Update torchir_passes.py remove_getattr_nodes() description
1 parent 3189e48 commit 7db5338

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

coremltools/converters/mil/frontend/torch/torchir_passes.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,8 @@ def forward(self, x):
231231

232232
def remove_getattr_nodes(graph: InternalTorchIRGraph) -> None:
233233
"""
234-
Remove the getattr nodes in the graph that are not output nodes
234+
Remove the getattr nodes in the graph
235+
If they are output nodes, convert them to constant nodes
235236
"""
236237

237238
new_nodes = []

0 commit comments

Comments
 (0)