nnlib
GPU-accelerated, C/C++ neural network library.
Public Member Functions | List of all members
AddBroadcast Class Reference
Inheritance diagram for AddBroadcast:
Function< sTensor, sTensor > BackwardFunction

Public Member Functions

sTensor forwardFn (const sTensor &a, const sTensor &b) override
 
std::vector< sTensor > backwardFn (sTensor grad) override
 
- Public Member Functions inherited from Function< sTensor, sTensor >
sTensor forward (const Types &... args)
 
std::vector< sTensor > backward (sTensor grad) override
 
virtual sTensor forwardFn (const Types &... args)=0
 

Additional Inherited Members

- Public Attributes inherited from BackwardFunction
std::vector< sTensor > parents
 

The documentation for this class was generated from the following files: