darknet  v3
Functions
activation_kernels.cu File Reference
#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "activations.h"
#include "cuda.h"
Include dependency graph for activation_kernels.cu:

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)
 

Function Documentation

◆ activate_array_gpu()

void activate_array_gpu ( float *  x,
int  n,
ACTIVATION  a 
)

Definition at line 196 of file activation_kernels.cu.

◆ activate_array_kernel()

__global__ void activate_array_kernel ( float *  x,
int  n,
ACTIVATION  a 
)

Definition at line 184 of file activation_kernels.cu.

◆ activate_kernel()

__device__ float activate_kernel ( float  x,
ACTIVATION  a 
)

Definition at line 79 of file activation_kernels.cu.

◆ binary_activate_array_gpu()

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.

◆ binary_activate_array_kernel()

__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.

◆ binary_gradient_array_gpu()

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.

◆ binary_gradient_array_kernel()

__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.

◆ elu_activate_kernel()

__device__ float elu_activate_kernel ( float  x)

Definition at line 33 of file activation_kernels.cu.

◆ elu_gradient_kernel()

__device__ float elu_gradient_kernel ( float  x)

Definition at line 66 of file activation_kernels.cu.

◆ gradient_array_gpu()

void gradient_array_gpu ( float *  x,
int  n,
ACTIVATION  a,
float *  delta 
)

Definition at line 202 of file activation_kernels.cu.

◆ gradient_array_kernel()

__global__ void gradient_array_kernel ( float *  x,
int  n,
ACTIVATION  a,
float *  delta 
)

Definition at line 190 of file activation_kernels.cu.

◆ gradient_kernel()

__device__ float gradient_kernel ( float  x,
ACTIVATION  a 
)

Definition at line 114 of file activation_kernels.cu.

◆ hardtan_activate_kernel()

__device__ float hardtan_activate_kernel ( float  x)

Definition at line 23 of file activation_kernels.cu.

◆ hardtan_gradient_kernel()

__device__ float hardtan_gradient_kernel ( float  x)

Definition at line 53 of file activation_kernels.cu.

◆ leaky_activate_kernel()

__device__ float leaky_activate_kernel ( float  x)

Definition at line 37 of file activation_kernels.cu.

◆ leaky_gradient_kernel()

__device__ float leaky_gradient_kernel ( float  x)

Definition at line 70 of file activation_kernels.cu.

◆ lhtan_activate_kernel()

__device__ float lhtan_activate_kernel ( float  x)

Definition at line 11 of file activation_kernels.cu.

◆ lhtan_gradient_kernel()

__device__ float lhtan_gradient_kernel ( float  x)

Definition at line 17 of file activation_kernels.cu.

◆ linear_activate_kernel()

__device__ float linear_activate_kernel ( float  x)

Definition at line 29 of file activation_kernels.cu.

◆ linear_gradient_kernel()

__device__ float linear_gradient_kernel ( float  x)

Definition at line 58 of file activation_kernels.cu.

◆ loggy_activate_kernel()

__device__ float loggy_activate_kernel ( float  x)

Definition at line 31 of file activation_kernels.cu.

◆ loggy_gradient_kernel()

__device__ float loggy_gradient_kernel ( float  x)

Definition at line 60 of file activation_kernels.cu.

◆ logistic_activate_kernel()

__device__ float logistic_activate_kernel ( float  x)

Definition at line 30 of file activation_kernels.cu.

◆ logistic_gradient_kernel()

__device__ float logistic_gradient_kernel ( float  x)

Definition at line 59 of file activation_kernels.cu.

◆ plse_activate_kernel()

__device__ float plse_activate_kernel ( float  x)

Definition at line 39 of file activation_kernels.cu.

◆ plse_gradient_kernel()

__device__ float plse_gradient_kernel ( float  x)

Definition at line 72 of file activation_kernels.cu.

◆ ramp_activate_kernel()

__device__ float ramp_activate_kernel ( float  x)

Definition at line 36 of file activation_kernels.cu.

◆ ramp_gradient_kernel()

__device__ float ramp_gradient_kernel ( float  x)

Definition at line 69 of file activation_kernels.cu.

◆ relie_activate_kernel()

__device__ float relie_activate_kernel ( float  x)

Definition at line 35 of file activation_kernels.cu.

◆ relie_gradient_kernel()

__device__ float relie_gradient_kernel ( float  x)

Definition at line 68 of file activation_kernels.cu.

◆ relu_activate_kernel()

__device__ float relu_activate_kernel ( float  x)

Definition at line 32 of file activation_kernels.cu.

◆ relu_gradient_kernel()

__device__ float relu_gradient_kernel ( float  x)

Definition at line 65 of file activation_kernels.cu.

◆ selu_activate_kernel()

__device__ float selu_activate_kernel ( float  x)

Definition at line 34 of file activation_kernels.cu.

◆ selu_gradient_kernel()

__device__ float selu_gradient_kernel ( float  x)

Definition at line 67 of file activation_kernels.cu.

◆ stair_activate_kernel()

__device__ float stair_activate_kernel ( float  x)

Definition at line 45 of file activation_kernels.cu.

◆ stair_gradient_kernel()

__device__ float stair_gradient_kernel ( float  x)

Definition at line 73 of file activation_kernels.cu.

◆ tanh_activate_kernel()

__device__ float tanh_activate_kernel ( float  x)

Definition at line 38 of file activation_kernels.cu.

◆ tanh_gradient_kernel()

__device__ float tanh_gradient_kernel ( float  x)

Definition at line 71 of file activation_kernels.cu.