Migrate batch norm layer

This commit is contained in:
2025-11-21 23:24:14 +01:00
parent 5679dc0a50
commit fd4775faa4
11 changed files with 181 additions and 364 deletions

View File

@@ -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);