File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -101,10 +101,10 @@ __global__ void GPUROIPoolBackward(
101
101
int index = blockIdx .x * blockDim .x + threadIdx .x ;
102
102
int offset = blockDim .x * gridDim .x ;
103
103
for (int i = index; i < nthreads; i += offset) {
104
- int pw = index % pooled_width;
105
- int ph = (index / pooled_width) % pooled_height;
106
- int c = (index / pooled_width / pooled_height) % channels;
107
- int n = index / pooled_width / pooled_height / channels;
104
+ int pw = i % pooled_width;
105
+ int ph = (i / pooled_width) % pooled_height;
106
+ int c = (i / pooled_width / pooled_height) % channels;
107
+ int n = i / pooled_width / pooled_height / channels;
108
108
109
109
int roi_batch_ind = roi_batch_id_data[n];
110
110
int input_offset = (roi_batch_ind * channels + c) * height * width;
You can’t perform that action at this time.
0 commit comments