feat: Softmax works with any number of dimensions
Showing
- aidge_export_cpp/kernels/softmax.hpp 32 additions, 34 deletionsaidge_export_cpp/kernels/softmax.hpp
- aidge_export_cpp/operators.py 24 additions, 0 deletionsaidge_export_cpp/operators.py
- aidge_export_cpp/templates/configuration/softmax_config.jinja 3 additions, 1 deletion...e_export_cpp/templates/configuration/softmax_config.jinja
- aidge_export_cpp/templates/kernel_forward/softmax_forward.jinja 4 additions, 8 deletions...export_cpp/templates/kernel_forward/softmax_forward.jinja
- aidge_export_cpp/unit_tests/test_export.py 24 additions, 0 deletionsaidge_export_cpp/unit_tests/test_export.py
Please register or sign in to comment