1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#include <_kernels.cu>
#include <_murmur3.cu>
void
gpu_maxout(float* best, int* which,
const float* cands, int B, int O, int P)
{
maxout<<<B/16,16>>>(best, which, cands, B, O, P);
}
void
gpu_mean_pool(float* means,
const float* X, const int* lengths, int B, int T, int O)
{
mean_pool<<<B, 1>>>(means, X, lengths, B, T, O);
}
void
gpu_max_pool(float* maxes, int* which,
const float* X, const int* lengths, int B, int T, int O)
{
max_pool<<<B, 1>>>(maxes, which, X, lengths, B, T, O);
}
void
gpu_sum_pool(float* sums,
const float* X, const int* lengths, int B, int T, int O)
{
sum_pool<<<B, 1>>>(sums, X, lengths, B, T, O);
}
void
gpu_backprop_mean_pool(float* dX, const float* d_means, const int* lengths, int B, int T, int O)
{
backprop_mean_pool<<<B, 1>>>(dX, d_means, lengths, B, T, O);
}
void
gpu_backprop_sum_pool(float* dX, const float* d_sums, const int* lengths, int B, int T, int O)
{
backprop_sum_pool<<<B, 1>>>(dX, d_sums, lengths, B, T, O);
}
void
gpu_backprop_max_pool(float* dX, const float* d_maxes, const int* which,
const int* lengths, int B, int T, int O)
{
backprop_max_pool<<<B, 1>>>(dX, d_maxes, which, lengths, B, T, O);
}
void
gpu_hash_data(char* dest,
const char* src, size_t out_size, size_t in_size, size_t n_items, uint32_t seed)
{
hash_data<<<n_items,1>>>(dest, src, out_size, in_size, n_items, seed);
}
|
