File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change 2222from executorch .backends .arm .tosa_specification import TosaSpecification
2323from executorch .backends .arm .tosa_utils import tosa_shape
2424from torch .fx import Node
25- from tosa .NanPropagationMode import NanPropagationMode # type: ignore[import-untyped]
2625
2726
2827@register_node_visitor
@@ -112,6 +111,7 @@ def define_node(
112111 ) -> None :
113112
114113 import serializer .tosa_serializer as ts # type: ignore
114+ from tosa import NanPropagationMode # type: ignore[import-untyped]
115115
116116 if inputs [0 ].dtype != inputs [1 ].dtype and inputs [0 ].dtype != output .dtype :
117117 raise TypeError (
Original file line number Diff line number Diff line change 2020from executorch .backends .arm .tosa_specification import TosaSpecification
2121from executorch .backends .arm .tosa_utils import tosa_shape
2222from torch .fx import Node
23- from tosa .NanPropagationMode import NanPropagationMode # type: ignore[import-untyped]
2423
2524
2625@register_node_visitor
@@ -110,6 +109,7 @@ def define_node(
110109 ) -> None :
111110
112111 import serializer .tosa_serializer as ts # type: ignore
112+ from tosa import NanPropagationMode # type: ignore[import-untyped]
113113
114114 if inputs [0 ].dtype != inputs [1 ].dtype and inputs [0 ].dtype != output .dtype :
115115 raise TypeError (
You can’t perform that action at this time.
0 commit comments