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
magmadnn
Definition:
addop.cpp:11
include
memory
memory_internal_device.h
Generated by
1.8.13