MobileNet operators
Context
close #5 (closed)
Complete the missing cuda operators' backends to run a MobileNetV2 inference:
- BatchNorm
- Reshape
- Add
- GlobalAveragePooling
- Pad
- ConvDepthWise
Modified files
-
BatchNormImpl.hpp
,BatchNormImpl.cpp
andTest_BatchNormImpl.cpp
, add BatchNorm cuda Impl; -
ReshapeImpl.hpp
,ReshapeImpl.cpp
andTest_ReshapeImpl.cpp
, add Reshape cuda Impl; -
AddImpl.hpp
,AddImpl.cpp
andTest_AddImpl.cpp
, add Add cuda Impl; -
GlobalAveragePoolingImpl.hpp
,GlobalAveragePoolingImpl.cpp
andTest_GlobalAveragePoolingImpl.cpp
, add GlobalAveragePooling cuda Impl; -
PadImpl.hpp
,PadImpl.cpp
andTest_PadImpl.cpp
, add Pad cuda Impl; -
ConvImpl.hpp
,ConvImpl.cpp
andTest_ConvDepthWiseImpl.cpp
, add ConvDepthWise cuda Impl;
TODO
-
DONE : Add "BatchNorm" Impl -
DONE : Add "Reshape" Impl -
DONE : Add "Add" Impl -
DONE : Add "GlobalAveragePooling" Impl -
DONE : Add "Pad" Impl -
DONE : Add "ConvDepthWise" Impl
Edited by Maxence Naud