Skip to content
Snippets Groups Projects

MobileNet operators

Merged Houssem ROUIS requested to merge hrouis/aidge_backend_cuda:mobilenet_operators into dev

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 and Test_BatchNormImpl.cpp, add BatchNorm cuda Impl;
  • ReshapeImpl.hpp, ReshapeImpl.cpp and Test_ReshapeImpl.cpp, add Reshape cuda Impl;
  • AddImpl.hpp, AddImpl.cpp and Test_AddImpl.cpp, add Add cuda Impl;
  • GlobalAveragePoolingImpl.hpp, GlobalAveragePoolingImpl.cpp and Test_GlobalAveragePoolingImpl.cpp, add GlobalAveragePooling cuda Impl;
  • PadImpl.hpp, PadImpl.cpp and Test_PadImpl.cpp, add Pad cuda Impl;
  • ConvImpl.hpp, ConvImpl.cpp and Test_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

Merge request reports

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading