MagmaDNN  1.0
c++NeuralNetworkFramework
fill_internal.h
Go to the documentation of this file.
1 
9 #pragma once
10 
11 #include <vector>
12 #include <random>
13 #include "utilities_internal.h"
14 #include "memory/memorymanager.h"
15 
16 #if defined(_HAS_CUDA_)
17 #include <cuda.h>
18 #include <curand.h>
19 #endif
20 
21 namespace magmadnn {
22 namespace internal {
23 
24 #if defined(_HAS_CUDA_)
25 
30 template <typename T>
31 void fill_constant_device(MemoryManager<T> &m, T val);
32 #endif
33 
39 template <typename T>
40 void fill_uniform(MemoryManager<T> &m, const std::vector<T>& params);
41 
47 template <typename T>
48 void fill_glorot(MemoryManager<T> &m, const std::vector<T>& params);
49 
55 template <typename T>
56 void fill_mask(MemoryManager<T> &m, const std::vector<T>& params);
57 
65 template <typename T>
66 void fill_diagonal(MemoryManager<T> &m, const std::vector<T>& params);
67 
73 template <typename T>
74 void fill_constant(MemoryManager<T> &m, const std::vector<T>& params);
75 
76 } // namespace internal
77 } // namespace magmadnn
magmadnn::internal::fill_constant
void fill_constant(MemoryManager< T > &m, const std::vector< T > &params)
Definition: fill_internal_host.cpp:310
magmadnn
Definition: addop.cpp:11
magmadnn::internal::fill_mask
void fill_mask(MemoryManager< T > &m, const std::vector< T > &params)
Definition: fill_internal_host.cpp:185
magmadnn::internal::fill_uniform
void fill_uniform(MemoryManager< T > &m, const std::vector< T > &params)
Definition: fill_internal_host.cpp:16
magmadnn::internal::fill_glorot
void fill_glorot(MemoryManager< T > &m, const std::vector< T > &params)
Definition: fill_internal_host.cpp:108
utilities_internal.h
magmadnn::internal::fill_diagonal
void fill_diagonal(MemoryManager< T > &m, const std::vector< T > &params)
Definition: fill_internal_host.cpp:268