diff options
| -rw-r--r-- | src/CuTNetLib/cuComponent.h | 25 |
1 files changed, 10 insertions, 15 deletions
diff --git a/src/CuTNetLib/cuComponent.h b/src/CuTNetLib/cuComponent.h index 7b92cb9..8d0d810 100644 --- a/src/CuTNetLib/cuComponent.h +++ b/src/CuTNetLib/cuComponent.h @@ -182,10 +182,16 @@ namespace TNet { virtual void WriteToStream(std::ostream& rOut) { } /// Public wrapper for PropagateFnc - void PropagateF(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y); + virtual void PropagateF(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y) + { + PropagateFnc(X,Y); + } + /// Public wrapper for BackpropagateFnc - void BackpropagateF(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y); - + virtual void BackpropagateF(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y) + { + BackpropagateFnc(X,Y); + } /////////////////////////////////////////////////////////////// // Nonpublic member functions used to update data outputs @@ -342,18 +348,7 @@ namespace TNet { mNOutputs=nOutputs; } - inline void - CuComponent:: - PropagateF(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y) - { - PropagateFnc(X,Y); - } - inline void - CuComponent:: - BackpropagateF(const CuMatrix<BaseFloat>& X, CuMatrix<BaseFloat>& Y) - { - BackpropagateFnc(X,Y); - } + ////////////////////////////////////////////////////////////////////////// |