src/network.h
@@ -21,7 +21,7 @@ float *workspace; int n; int batch; uint64_t *seen; int *seen; float epoch; int subdivisions; float momentum; @@ -64,6 +64,11 @@ #ifdef GPU float **input_gpu; float **truth_gpu; float **input16_gpu; float **output16_gpu; size_t *max_input16_size; size_t *max_output16_size; int wait_stream; #endif } network;