Searched refs:input16_gpu (Results 1 – 4 of 4) sorted by relevance
440 if (*state.net.input16_gpu) cuda_free(*state.net.input16_gpu); in forward_convolutional_layer_gpu()442 … *state.net.input16_gpu = (float *)cuda_make_f16_from_f32_array(NULL, *state.net.max_input16_size); in forward_convolutional_layer_gpu()444 float *input16 = *state.net.input16_gpu; in forward_convolutional_layer_gpu()683 if (*state.net.input16_gpu) cuda_free(*state.net.input16_gpu); in backward_convolutional_layer_gpu()685 … *state.net.input16_gpu = (float *)cuda_make_f16_from_f32_array(NULL, *state.net.max_input16_size); in backward_convolutional_layer_gpu()687 float *input16 = *state.net.input16_gpu; in backward_convolutional_layer_gpu()
253 net.input16_gpu = (float**)xcalloc(1, sizeof(float*)); in make_network()1161 if (*net.input16_gpu) cuda_free(*net.input16_gpu); in free_network()1163 if (net.input16_gpu) free(net.input16_gpu); in free_network()
1594 …CHECK_CUDA(cudaMalloc((void **)net.input16_gpu, *net.max_input16_size * sizeof(short))); //sizeof(… in parse_network_cfg_custom()
752 float **input16_gpu; member