mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Migrate batch norm layer
This commit is contained in:
@@ -45,6 +45,11 @@ public:
|
||||
|
||||
void zero();
|
||||
|
||||
template <typename T>
|
||||
void fill(T value) {
|
||||
backend->fill(*this, value);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
void set_data(T *data) {
|
||||
backend->copy_to_device(*this, data, total_size);
|
||||
|
||||
Reference in New Issue
Block a user