From 62235e9aa3d0c15d87d49bf340625d075cba3e65 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Sat, 19 Nov 2016 05:51:36 +0000 Subject: [PATCH] cpu batch norm works --- src/utils.h | 18 ++++++++++++++++-- 1 files changed, 16 insertions(+), 2 deletions(-) diff --git a/src/utils.h b/src/utils.h index 1b9ba08..bbc6765 100644 --- a/src/utils.h +++ b/src/utils.h @@ -5,16 +5,22 @@ #include "list.h" #define SECRET_NUM -1234 +#define TWO_PI 6.2831853071795864769252866 +int *read_map(char *filename); void shuffle(void *arr, size_t n, size_t size); void sorta_shuffle(void *arr, size_t n, size_t size, size_t sections); void free_ptrs(void **ptrs, int n); char *basecfg(char *cfgfile); int alphanum_to_int(char c); char int_to_alphanum(int i); +int read_int(int fd); +void write_int(int fd, int n); void read_all(int fd, char *buffer, size_t bytes); void write_all(int fd, char *buffer, size_t bytes); -char *find_replace(char *str, char *orig, char *rep); +int read_all_fail(int fd, char *buffer, size_t bytes); +int write_all_fail(int fd, char *buffer, size_t bytes); +void find_replace(char *str, char *orig, char *rep, char *output); void error(const char *s); void malloc_error(); void file_error(char *s); @@ -32,19 +38,27 @@ void translate_array(float *a, int n, float s); int max_index(float *a, int n); float constrain(float min, float max, float a); +int constrain_int(int a, int min, int max); float mse_array(float *a, int n); float rand_normal(); -float rand_uniform(); +size_t rand_size_t(); +float rand_uniform(float min, float max); +float rand_scale(float s); +int rand_int(int min, int max); float sum_array(float *a, int n); float mean_array(float *a, int n); +void mean_arrays(float **a, int n, int els, float *avg); float variance_array(float *a, int n); float mag_array(float *a, int n); +float dist_array(float *a, float *b, int n, int sub); float **one_hot_encode(float *a, int n, int k); float sec(clock_t clocks); int find_int_arg(int argc, char **argv, char *arg, int def); float find_float_arg(int argc, char **argv, char *arg, float def); int find_arg(int argc, char* argv[], char *arg); char *find_char_arg(int argc, char **argv, char *arg, char *def); +int sample_array(float *a, int n); +void print_statistics(float *a, int n); #endif -- Gitblit v1.10.0