Skip to content

Commit 7abecdf

Browse files
committed
handle softplus in transpose optimizer, needed for mish
1 parent d3dd7f0 commit 7abecdf

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@ def _initialize_handlers(self):
188188
"Sum": self._sum_handler,
189189
"Slice": self._slice_handler,
190190
"Split": self._split_handler,
191+
"Softplus": self._simple_through_handler,
191192
"Squeeze": self._squeeze_handler,
192193
"Sub": self._sub_handler,
193194
"Tanh": self._simple_through_handler,

0 commit comments

Comments
 (0)