![]() |
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.
1.8.13