CVPR Experiments

Joseph Redmon

2015-11-04 8fd18add6e060a433629fae3fa2a7ef75df4644e

 src/softmax_layer_kernels.cu

@@ -4,8 +4,6 @@
#include "blas.h"
}

#define BLOCK 256

__global__ void forward_softmax_layer_kernel(int n, int batch, float *input, float *output)
{
    int b = (blockIdx.x + blockIdx.y*gridDim.x) * blockDim.x + threadIdx.x;