mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Refactor Backend and Layer interfaces
This commit is contained in:
@@ -2,6 +2,7 @@
|
||||
|
||||
#include "backend/tensor.hpp"
|
||||
#include "backend/backend.hpp"
|
||||
#include "layers/layer.hpp"
|
||||
|
||||
namespace CUDANet::Layers {
|
||||
|
||||
@@ -19,7 +20,7 @@ enum ActivationType { SIGMOID, RELU, SOFTMAX, NONE };
|
||||
* @brief Utility class that performs activation
|
||||
*
|
||||
*/
|
||||
class Activation {
|
||||
class Activation : Layer {
|
||||
public:
|
||||
|
||||
Activation() = default;
|
||||
|
||||
Reference in New Issue
Block a user