Skip to content

Draft: Learning with backend cuda

Houssem ROUIS requested to merge hrouis/aidge_core:dev into dev

Context

Suppport Learning with backend_cuda

Modified files

  • DataProvider.hpp , DataProvider.cpp and pybind_DataProvider.cpp, add backend member to DataProvider;

TODO

  • Support backend_cuda in DataProvider
  • Add Argmax operator for Accuracy computation
  • Add And operator for Accuracy computation
  • Add ReduceSum operator for Accuracy computation
Edited by Houssem ROUIS

Merge request reports