Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
UltralyticsAssistant committed Dec 10, 2024
1 parent 0318396 commit da66046
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions onnxslim/third_party/onnx_graphsurgeon/ir/node.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from dataclasses import dataclass
from typing import Dict, List, Union

from onnxslim.third_party.onnx_graphsurgeon.ir.tensor import Constant, Tensor, Variable
from onnxslim.third_party.onnx_graphsurgeon.ir.tensor import Constant, Tensor
from onnxslim.third_party.onnx_graphsurgeon.logger import G_LOGGER
from onnxslim.third_party.onnx_graphsurgeon.util import misc

Expand Down Expand Up @@ -238,7 +238,7 @@ def feeds(self):
feeds.extend(input if feed.op == "Split" else feed for feed in input.inputs)
return feeds

def replace_all_uses_with(self, node: Union['Node', 'Tensor'], input_var_idx=0, output_var_idx=0):
def replace_all_uses_with(self, node: Union["Node", "Tensor"], input_var_idx=0, output_var_idx=0):
"""Replace all uses of this node with the given node."""
if isinstance(node, Node):
input_var = node.outputs[output_var_idx]
Expand Down

0 comments on commit da66046

Please sign in to comment.