![]() |
darknet
v3
|
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "activations.h"
#include "cuda.h"
Go to the source code of this file.
Functions | |
__device__ float | lhtan_activate_kernel (float x) |
__device__ float | lhtan_gradient_kernel (float x) |
__device__ float | hardtan_activate_kernel (float x) |
__device__ float | linear_activate_kernel (float x) |
__device__ float | logistic_activate_kernel (float x) |
__device__ float | loggy_activate_kernel (float x) |
__device__ float | relu_activate_kernel (float x) |
__device__ float | elu_activate_kernel (float x) |
__device__ float | selu_activate_kernel (float x) |
__device__ float | relie_activate_kernel (float x) |
__device__ float | ramp_activate_kernel (float x) |
__device__ float | leaky_activate_kernel (float x) |
__device__ float | tanh_activate_kernel (float x) |
__device__ float | plse_activate_kernel (float x) |
__device__ float | stair_activate_kernel (float x) |
__device__ float | hardtan_gradient_kernel (float x) |
__device__ float | linear_gradient_kernel (float x) |
__device__ float | logistic_gradient_kernel (float x) |
__device__ float | loggy_gradient_kernel (float x) |
__device__ float | relu_gradient_kernel (float x) |
__device__ float | elu_gradient_kernel (float x) |
__device__ float | selu_gradient_kernel (float x) |
__device__ float | relie_gradient_kernel (float x) |
__device__ float | ramp_gradient_kernel (float x) |
__device__ float | leaky_gradient_kernel (float x) |
__device__ float | tanh_gradient_kernel (float x) |
__device__ float | plse_gradient_kernel (float x) |
__device__ float | stair_gradient_kernel (float x) |
__device__ float | activate_kernel (float x, ACTIVATION a) |
__device__ float | gradient_kernel (float x, ACTIVATION a) |
__global__ void | binary_gradient_array_kernel (float *x, float *dy, int n, int s, BINARY_ACTIVATION a, float *dx) |
void | binary_gradient_array_gpu (float *x, float *dx, int n, int size, BINARY_ACTIVATION a, float *y) |
__global__ void | binary_activate_array_kernel (float *x, int n, int s, BINARY_ACTIVATION a, float *y) |
void | binary_activate_array_gpu (float *x, int n, int size, BINARY_ACTIVATION a, float *y) |
__global__ void | activate_array_kernel (float *x, int n, ACTIVATION a) |
__global__ void | gradient_array_kernel (float *x, int n, ACTIVATION a, float *delta) |
void | activate_array_gpu (float *x, int n, ACTIVATION a) |
void | gradient_array_gpu (float *x, int n, ACTIVATION a, float *delta) |
void activate_array_gpu | ( | float * | x, |
int | n, | ||
ACTIVATION | a | ||
) |
Definition at line 196 of file activation_kernels.cu.
__global__ void activate_array_kernel | ( | float * | x, |
int | n, | ||
ACTIVATION | a | ||
) |
Definition at line 184 of file activation_kernels.cu.
__device__ float activate_kernel | ( | float | x, |
ACTIVATION | a | ||
) |
Definition at line 79 of file activation_kernels.cu.
void binary_activate_array_gpu | ( | float * | x, |
int | n, | ||
int | size, | ||
BINARY_ACTIVATION | a, | ||
float * | y | ||
) |
Definition at line 178 of file activation_kernels.cu.
__global__ void binary_activate_array_kernel | ( | float * | x, |
int | n, | ||
int | s, | ||
BINARY_ACTIVATION | a, | ||
float * | y | ||
) |
Definition at line 168 of file activation_kernels.cu.
void binary_gradient_array_gpu | ( | float * | x, |
float * | dx, | ||
int | n, | ||
int | size, | ||
BINARY_ACTIVATION | a, | ||
float * | y | ||
) |
Definition at line 163 of file activation_kernels.cu.
__global__ void binary_gradient_array_kernel | ( | float * | x, |
float * | dy, | ||
int | n, | ||
int | s, | ||
BINARY_ACTIVATION | a, | ||
float * | dx | ||
) |
Definition at line 149 of file activation_kernels.cu.
__device__ float elu_activate_kernel | ( | float | x | ) |
Definition at line 33 of file activation_kernels.cu.
__device__ float elu_gradient_kernel | ( | float | x | ) |
Definition at line 66 of file activation_kernels.cu.
void gradient_array_gpu | ( | float * | x, |
int | n, | ||
ACTIVATION | a, | ||
float * | delta | ||
) |
Definition at line 202 of file activation_kernels.cu.
__global__ void gradient_array_kernel | ( | float * | x, |
int | n, | ||
ACTIVATION | a, | ||
float * | delta | ||
) |
Definition at line 190 of file activation_kernels.cu.
__device__ float gradient_kernel | ( | float | x, |
ACTIVATION | a | ||
) |
Definition at line 114 of file activation_kernels.cu.
__device__ float hardtan_activate_kernel | ( | float | x | ) |
Definition at line 23 of file activation_kernels.cu.
__device__ float hardtan_gradient_kernel | ( | float | x | ) |
Definition at line 53 of file activation_kernels.cu.
__device__ float leaky_activate_kernel | ( | float | x | ) |
Definition at line 37 of file activation_kernels.cu.
__device__ float leaky_gradient_kernel | ( | float | x | ) |
Definition at line 70 of file activation_kernels.cu.
__device__ float lhtan_activate_kernel | ( | float | x | ) |
Definition at line 11 of file activation_kernels.cu.
__device__ float lhtan_gradient_kernel | ( | float | x | ) |
Definition at line 17 of file activation_kernels.cu.
__device__ float linear_activate_kernel | ( | float | x | ) |
Definition at line 29 of file activation_kernels.cu.
__device__ float linear_gradient_kernel | ( | float | x | ) |
Definition at line 58 of file activation_kernels.cu.
__device__ float loggy_activate_kernel | ( | float | x | ) |
Definition at line 31 of file activation_kernels.cu.
__device__ float loggy_gradient_kernel | ( | float | x | ) |
Definition at line 60 of file activation_kernels.cu.
__device__ float logistic_activate_kernel | ( | float | x | ) |
Definition at line 30 of file activation_kernels.cu.
__device__ float logistic_gradient_kernel | ( | float | x | ) |
Definition at line 59 of file activation_kernels.cu.
__device__ float plse_activate_kernel | ( | float | x | ) |
Definition at line 39 of file activation_kernels.cu.
__device__ float plse_gradient_kernel | ( | float | x | ) |
Definition at line 72 of file activation_kernels.cu.
__device__ float ramp_activate_kernel | ( | float | x | ) |
Definition at line 36 of file activation_kernels.cu.
__device__ float ramp_gradient_kernel | ( | float | x | ) |
Definition at line 69 of file activation_kernels.cu.
__device__ float relie_activate_kernel | ( | float | x | ) |
Definition at line 35 of file activation_kernels.cu.
__device__ float relie_gradient_kernel | ( | float | x | ) |
Definition at line 68 of file activation_kernels.cu.
__device__ float relu_activate_kernel | ( | float | x | ) |
Definition at line 32 of file activation_kernels.cu.
__device__ float relu_gradient_kernel | ( | float | x | ) |
Definition at line 65 of file activation_kernels.cu.
__device__ float selu_activate_kernel | ( | float | x | ) |
Definition at line 34 of file activation_kernels.cu.
__device__ float selu_gradient_kernel | ( | float | x | ) |
Definition at line 67 of file activation_kernels.cu.
__device__ float stair_activate_kernel | ( | float | x | ) |
Definition at line 45 of file activation_kernels.cu.
__device__ float stair_gradient_kernel | ( | float | x | ) |
Definition at line 73 of file activation_kernels.cu.
__device__ float tanh_activate_kernel | ( | float | x | ) |
Definition at line 38 of file activation_kernels.cu.
__device__ float tanh_gradient_kernel | ( | float | x | ) |
Definition at line 71 of file activation_kernels.cu.