Skip to content

Commit 4671158

Browse files
committed
[Test] switch is_finn_op to is_custom_op
1 parent 7bfc299 commit 4671158

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/transformation/test_channelslast.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
from qonnx.transformation.infer_shapes import InferShapes
4747
from qonnx.transformation.make_input_chanlast import MakeInputChannelsLast
4848
from qonnx.transformation.quant_constant_folding import FoldTransposeIntoQuantInit
49-
from qonnx.util.basic import is_finn_op
49+
from qonnx.util.basic import is_custom_op
5050
from qonnx.util.test import download_model, get_golden_in_and_output, test_model_details
5151
from qonnx.util.to_channels_last import to_channels_last
5252

@@ -125,7 +125,7 @@ def analysis_test_for_left_transposes(model, test_model, make_input_channels_las
125125
def verify_all_nodes(model):
126126
result = dict()
127127
for n in model.graph.node:
128-
if is_finn_op(n.domain):
128+
if is_custom_op(n.domain):
129129
n_instance = getCustomOp(n)
130130
verify_result = n_instance.verify_node()
131131
result[n.name] = verify_result

0 commit comments

Comments
 (0)