-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathkernel.h
More file actions
25 lines (15 loc) · 775 Bytes
/
kernel.h
File metadata and controls
25 lines (15 loc) · 775 Bytes
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
#ifndef DEF_KERNEL
#define DEF_KERNEL
#include "RandomForest.h"
#include "GPUAdapter.h"
/*__device__*/
float gpuGetValue (float *gpuFeatures, uint8_t channel,
int16_t x, int16_t y, int16_t w, int16_t h);
/*__device__*/
float gpuGetValueIntegral (float *gpuFeaturesIntegral, uint8_t channel,
int16_t x1, int16_t y1, int16_t x2, int16_t y2, int16_t w, int16_t h);
/*__device__*/
SplitResult split(SplitData<float> splitData, Sample<float> &sample, int16_t w, int16_t h, int16_t w_i, int16_t h_i, float *gpuFeatures, float *gpuFeaturesIntegral);
/*__device__*/
void predict(int *returnStartHistTab, int *returnCountHistTab, ANode* tree, int16_t w, int16_t h, int16_t w_i, int16_t h_i, float* features, float* features_integral, Sample<float> &sample);
#endif