AxisCommunications / onnx-to-keras

Convert onnx models exported from pytorch to tensorflow keras models with focus on performace and highleve compatibility.
MIT License
25 stars 13 forks source link

Added op_softmax, #4 #6

Closed aitikgupta closed 3 years ago

aitikgupta commented 4 years ago

Added the Softmax layer support, referenced at #4