Skip to content

Commit ed02243

Browse files
Zyrinatsogias
andauthored
Add support for Erf and Selu in transpose_optimizer.py. (#2311)
Signed-off-by: Alexis Tsogias <a.tsogias@cellumation.com> Co-authored-by: Alexis Tsogias <a.tsogias@cellumation.com>
1 parent 358b519 commit ed02243

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -201,6 +201,7 @@ def _initialize_handlers(self):
201201
"Clip": self._simple_through_handler,
202202
"Concat": self._concat_handler,
203203
"Elu": self._simple_through_handler,
204+
"Erf": self._simple_through_handler,
204205
"Exp": self._simple_through_handler,
205206
"Identity": self._identity_handler,
206207
"LeakyRelu": self._simple_through_handler,
@@ -222,6 +223,7 @@ def _initialize_handlers(self):
222223
"ReduceSum": self._reducesum_handler,
223224
"ReduceSumSquare": self._reduce_handler,
224225
"Relu": self._simple_through_handler,
226+
"Selu": self._simple_through_handler,
225227
"Shape": self._shape_handler,
226228
"Sigmoid": self._simple_through_handler,
227229
"Softmax": self._softmax_handler,

0 commit comments

Comments
 (0)