mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 22:34:22 +00:00
Add dtype parameter to layer constructors
This commit is contained in:
@@ -16,6 +16,8 @@ namespace CUDANet {
|
||||
*
|
||||
*/
|
||||
class Layer {
|
||||
protected:
|
||||
CUDANet::DType dtype;
|
||||
public:
|
||||
|
||||
virtual ~Layer(){};
|
||||
@@ -39,4 +41,4 @@ class Layer {
|
||||
virtual size_t get_biases_size() = 0;
|
||||
};
|
||||
|
||||
} // namespace CUDANet::Layers
|
||||
} // namespace CUDANet
|
||||
|
||||
Reference in New Issue
Block a user