Draft: Learning with backend cuda
Context
Suppport Learning with backend_cuda
Modified files
-
DataProvider.hpp
,DataProvider.cpp
andpybind_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