add half_precision support
parent
ac87b48e
No related branches found
No related tags found
Checking pipeline status
Showing
- include/aidge/backend/cuda/operator/FCImpl_CUDA_kernels.hpp 13 additions, 7 deletionsinclude/aidge/backend/cuda/operator/FCImpl_CUDA_kernels.hpp
- include/aidge/backend/cuda/utils/CudaUtils.hpp 25 additions, 0 deletionsinclude/aidge/backend/cuda/utils/CudaUtils.hpp
- src/operator/AvgPoolingImpl.cpp 17 additions, 14 deletionssrc/operator/AvgPoolingImpl.cpp
- src/operator/ConvImpl.cpp 17 additions, 15 deletionssrc/operator/ConvImpl.cpp
- src/operator/FCImpl.cpp 21 additions, 16 deletionssrc/operator/FCImpl.cpp
- src/operator/FCImpl_CUDA_kernels.cu 42 additions, 42 deletionssrc/operator/FCImpl_CUDA_kernels.cu
- src/operator/MaxPoolingImpl.cpp 17 additions, 13 deletionssrc/operator/MaxPoolingImpl.cpp
- src/operator/ReLUImpl.cpp 5 additions, 8 deletionssrc/operator/ReLUImpl.cpp
- unit_tests/Test_AvgPoolingImpl.cpp 36 additions, 0 deletionsunit_tests/Test_AvgPoolingImpl.cpp
Loading
Please register or sign in to comment