darknet  v3
Functions
blas.h File Reference
#include "darknet.h"
Include dependency graph for blas.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

void flatten (float *x, int size, int layers, int batch, int forward)
 
void pm (int M, int N, float *A)
 
float * random_matrix (int rows, int cols)
 
void time_random_matrix (int TA, int TB, int m, int k, int n)
 
void reorg_cpu (float *x, int w, int h, int c, int batch, int stride, int forward, float *out)
 
void test_blas ()
 
void inter_cpu (int NX, float *X, int NY, float *Y, int B, float *OUT)
 
void deinter_cpu (int NX, float *X, int NY, float *Y, int B, float *OUT)
 
void mult_add_into_cpu (int N, float *X, float *Y, float *Z)
 
void const_cpu (int N, float ALPHA, float *X, int INCX)
 
void constrain_gpu (int N, float ALPHA, float *X, int INCX)
 
void pow_cpu (int N, float ALPHA, float *X, int INCX, float *Y, int INCY)
 
void mul_cpu (int N, float *X, int INCX, float *Y, int INCY)
 
int test_gpu_blas ()
 
void shortcut_cpu (int batch, int w1, int h1, int c1, float *add, int w2, int h2, int c2, float s1, float s2, float *out)
 
void mean_cpu (float *x, int batch, int filters, int spatial, float *mean)
 
void variance_cpu (float *x, float *mean, int batch, int filters, int spatial, float *variance)
 
void scale_bias (float *output, float *scales, int batch, int n, int size)
 
void backward_scale_cpu (float *x_norm, float *delta, int batch, int n, int size, float *scale_updates)
 
void mean_delta_cpu (float *delta, float *variance, int batch, int filters, int spatial, float *mean_delta)
 
void variance_delta_cpu (float *x, float *delta, float *mean, float *variance, int batch, int filters, int spatial, float *variance_delta)
 
void normalize_delta_cpu (float *x, float *mean, float *variance, float *mean_delta, float *variance_delta, int batch, int filters, int spatial, float *delta)
 
void l2normalize_cpu (float *x, float *dx, int batch, int filters, int spatial)
 
void smooth_l1_cpu (int n, float *pred, float *truth, float *delta, float *error)
 
void l2_cpu (int n, float *pred, float *truth, float *delta, float *error)
 
void l1_cpu (int n, float *pred, float *truth, float *delta, float *error)
 
void logistic_x_ent_cpu (int n, float *pred, float *truth, float *delta, float *error)
 
void softmax_x_ent_cpu (int n, float *pred, float *truth, float *delta, float *error)
 
void weighted_sum_cpu (float *a, float *b, float *s, int num, float *c)
 
void weighted_delta_cpu (float *a, float *b, float *s, float *da, float *db, float *ds, int n, float *dc)
 
void softmax (float *input, int n, float temp, int stride, float *output)
 
void softmax_cpu (float *input, int n, int batch, int batch_offset, int groups, int group_offset, int stride, float temp, float *output)
 
void upsample_cpu (float *in, int w, int h, int c, int batch, int stride, int forward, float scale, float *out)
 

Function Documentation

◆ backward_scale_cpu()

void backward_scale_cpu ( float *  x_norm,
float *  delta,
int  batch,
int  n,
int  size,
float *  scale_updates 
)

Definition at line 72 of file batchnorm_layer.c.

◆ const_cpu()

void const_cpu ( int  N,
float  ALPHA,
float *  X,
int  INCX 
)

Definition at line 160 of file blas.c.

◆ constrain_gpu()

void constrain_gpu ( int  N,
float  ALPHA,
float *  X,
int  INCX 
)

Definition at line 680 of file blas_kernels.cu.

◆ deinter_cpu()

void deinter_cpu ( int  NX,
float *  X,
int  NY,
float *  Y,
int  B,
float *  OUT 
)

Definition at line 196 of file blas.c.

◆ flatten()

void flatten ( float *  x,
int  size,
int  layers,
int  batch,
int  forward 
)

Definition at line 32 of file blas.c.

◆ inter_cpu()

void inter_cpu ( int  NX,
float *  X,
int  NY,
float *  Y,
int  B,
float *  OUT 
)

Definition at line 212 of file blas.c.

◆ l1_cpu()

void l1_cpu ( int  n,
float *  pred,
float *  truth,
float *  delta,
float *  error 
)

Definition at line 255 of file blas.c.

◆ l2_cpu()

void l2_cpu ( int  n,
float *  pred,
float *  truth,
float *  delta,
float *  error 
)

Definition at line 287 of file blas.c.

◆ l2normalize_cpu()

void l2normalize_cpu ( float *  x,
float *  dx,
int  batch,
int  filters,
int  spatial 
)

Definition at line 126 of file blas.c.

◆ logistic_x_ent_cpu()

void logistic_x_ent_cpu ( int  n,
float *  pred,
float *  truth,
float *  delta,
float *  error 
)

Definition at line 276 of file blas.c.

◆ mean_cpu()

void mean_cpu ( float *  x,
int  batch,
int  filters,
int  spatial,
float *  mean 
)

Definition at line 94 of file blas.c.

◆ mean_delta_cpu()

void mean_delta_cpu ( float *  delta,
float *  variance,
int  batch,
int  filters,
int  spatial,
float *  mean_delta 
)

Definition at line 87 of file batchnorm_layer.c.

◆ mul_cpu()

void mul_cpu ( int  N,
float *  X,
int  INCX,
float *  Y,
int  INCY 
)

Definition at line 166 of file blas.c.

◆ mult_add_into_cpu()

void mult_add_into_cpu ( int  N,
float *  X,
float *  Y,
float *  Z 
)

Definition at line 232 of file blas.c.

◆ normalize_delta_cpu()

void normalize_delta_cpu ( float *  x,
float *  mean,
float *  variance,
float *  mean_delta,
float *  variance_delta,
int  batch,
int  filters,
int  spatial,
float *  delta 
)

Definition at line 117 of file batchnorm_layer.c.

◆ pm()

void pm ( int  M,
int  N,
float *  A 
)

Definition at line 203 of file utils.c.

◆ pow_cpu()

void pow_cpu ( int  N,
float  ALPHA,
float *  X,
int  INCX,
float *  Y,
int  INCY 
)

Definition at line 172 of file blas.c.

◆ random_matrix()

float* random_matrix ( int  rows,
int  cols 
)

Definition at line 30 of file gemm.c.

◆ reorg_cpu()

void reorg_cpu ( float *  x,
int  w,
int  h,
int  c,
int  batch,
int  stride,
int  forward,
float *  out 
)

Definition at line 9 of file blas.c.

◆ scale_bias()

void scale_bias ( float *  output,
float *  scales,
int  batch,
int  n,
int  size 
)

Definition at line 423 of file convolutional_layer.c.

◆ shortcut_cpu()

void shortcut_cpu ( int  batch,
int  w1,
int  h1,
int  c1,
float *  add,
int  w2,
int  h2,
int  c2,
float  s1,
float  s2,
float *  out 
)

Definition at line 68 of file blas.c.

◆ smooth_l1_cpu()

void smooth_l1_cpu ( int  n,
float *  pred,
float *  truth,
float *  delta,
float *  error 
)

Definition at line 238 of file blas.c.

◆ softmax()

void softmax ( float *  input,
int  n,
float  temp,
int  stride,
float *  output 
)

Definition at line 305 of file blas.c.

◆ softmax_cpu()

void softmax_cpu ( float *  input,
int  n,
int  batch,
int  batch_offset,
int  groups,
int  group_offset,
int  stride,
float  temp,
float *  output 
)

Definition at line 324 of file blas.c.

◆ softmax_x_ent_cpu()

void softmax_x_ent_cpu ( int  n,
float *  pred,
float *  truth,
float *  delta,
float *  error 
)

Definition at line 265 of file blas.c.

◆ test_blas()

void test_blas ( )

◆ test_gpu_blas()

int test_gpu_blas ( )

◆ time_random_matrix()

void time_random_matrix ( int  TA,
int  TB,
int  m,
int  k,
int  n 
)

Definition at line 40 of file gemm.c.

◆ upsample_cpu()

void upsample_cpu ( float *  in,
int  w,
int  h,
int  c,
int  batch,
int  stride,
int  forward,
float  scale,
float *  out 
)

Definition at line 334 of file blas.c.

◆ variance_cpu()

void variance_cpu ( float *  x,
float *  mean,
int  batch,
int  filters,
int  spatial,
float *  variance 
)

Definition at line 110 of file blas.c.

◆ variance_delta_cpu()

void variance_delta_cpu ( float *  x,
float *  delta,
float *  mean,
float *  variance,
int  batch,
int  filters,
int  spatial,
float *  variance_delta 
)

Definition at line 102 of file batchnorm_layer.c.

◆ weighted_delta_cpu()

void weighted_delta_cpu ( float *  a,
float *  b,
float *  s,
float *  da,
float *  db,
float *  ds,
int  n,
float *  dc 
)

Definition at line 58 of file blas.c.

◆ weighted_sum_cpu()

void weighted_sum_cpu ( float *  a,
float *  b,
float *  s,
int  num,
float *  c 
)

Definition at line 50 of file blas.c.

Generated by   doxygen 1.8.13