add custom einsum to onnx_custom_objects_map

tensorleap · Dec 5, 2024 · 5b8911d · 5b8911d
1 parent 87a26a0
commit 5b8911d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnx2kerastl/customonnxlayer/__init__.py b/onnx2kerastl/customonnxlayer/__init__.py
@@ -3,6 +3,7 @@
 
 onnx_custom_objects_map = {
     "OnnxLSTM": OnnxLSTM,
+    "OnnxEinsumLayer": OnnxEinsumLayer
 }
 
 onnx_custom_layers = {