We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3189e48 commit 7db5338Copy full SHA for 7db5338
1 file changed
coremltools/converters/mil/frontend/torch/torchir_passes.py
@@ -231,7 +231,8 @@ def forward(self, x):
231
232
def remove_getattr_nodes(graph: InternalTorchIRGraph) -> None:
233
"""
234
- Remove the getattr nodes in the graph that are not output nodes
+ Remove the getattr nodes in the graph
235
+ If they are output nodes, convert them to constant nodes
236
237
238
new_nodes = []
0 commit comments