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 22
22
from executorch .backends .arm .tosa_specification import TosaSpecification
23
23
from executorch .backends .arm .tosa_utils import tosa_shape
24
24
from torch .fx import Node
25
- from tosa .NanPropagationMode import NanPropagationMode # type: ignore[import-untyped]
26
25
27
26
28
27
@register_node_visitor
@@ -112,6 +111,7 @@ def define_node(
112
111
) -> None :
113
112
114
113
import serializer .tosa_serializer as ts # type: ignore
114
+ from tosa import NanPropagationMode # type: ignore[import-untyped]
115
115
116
116
if inputs [0 ].dtype != inputs [1 ].dtype and inputs [0 ].dtype != output .dtype :
117
117
raise TypeError (
Original file line number Diff line number Diff line change 20
20
from executorch .backends .arm .tosa_specification import TosaSpecification
21
21
from executorch .backends .arm .tosa_utils import tosa_shape
22
22
from torch .fx import Node
23
- from tosa .NanPropagationMode import NanPropagationMode # type: ignore[import-untyped]
24
23
25
24
26
25
@register_node_visitor
@@ -110,6 +109,7 @@ def define_node(
110
109
) -> None :
111
110
112
111
import serializer .tosa_serializer as ts # type: ignore
112
+ from tosa import NanPropagationMode # type: ignore[import-untyped]
113
113
114
114
if inputs [0 ].dtype != inputs [1 ].dtype and inputs [0 ].dtype != output .dtype :
115
115
raise TypeError (
You can’t perform that action at this time.
0 commit comments