src/network.h @@ -64,6 +64,10 @@ #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;