Add MSE missing operators
Context
In order to use MSE with backend cuda
, the following operators need to be supported:
- Mul
- ReduceMean
- Pow
Solves #20 (closed)
Modified files
-
MulImpl.hpp
,MulImpl.cpp
andTest_MulImpl.cpp
, add Mul forward() and backward(); -
ReduceMeanImpl.hpp
,ReduceMeanImpl.cpp
andTest_ReduceMeanImpl.cpp
, add ReduceMean forward() and backward(); -
PowImpl.hpp
,PowImpl.cpp
andTest_PowImpl.cpp
, add Pow forward() and backward();
TODO
-
Mul forwad -
Mul backward -
ReduceMean forward -
ReduceMean backward -
Pow forward
Edited by Houssem ROUIS