MagmaDNN  1.0
c++NeuralNetworkFramework
memory_internal_device.h
Go to the documentation of this file.
1 
9 #pragma once
10 
11 #if defined(_HAS_CUDA_)
12 
13 namespace magmadnn {
14 namespace internal {
15 
22 template <typename T>
23 T get_device_array_element(T *arr, unsigned int idx);
24 
25 
32 template <typename T>
33 void set_device_array_element(T *arr, unsigned int idx, T val);
34 
35 } // namespace internal
36 } // namespace magmadnn
37 
38 #endif
Definition: addop.cpp:11