Skip to content

Commit

Permalink
Update deformable_convolution-inl.h
Browse files Browse the repository at this point in the history
  • Loading branch information
HaozhiQi authored Jul 3, 2017
1 parent c875eff commit aecdf52
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions faster_rcnn/operator_cxx/deformable_convolution-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -216,10 +216,7 @@ namespace mxnet {
in_data[conv::kOffset].dptr<DType>() + n*input_offset_dim_, in_data[conv::kData].shape_,
col_buffer.shape_, param_.kernel, param_.pad, param_.stride, param_.dilate, param_.num_deformable_group,
col_buffer.dptr<DType>());

im2col(s, in_data[conv::kData].dptr<DType>() + n*input_dim_, in_data[conv::kData].shape_,
col_buffer.shape_, param_.kernel, param_.pad, param_.stride, param_.dilate,
col_buffer.dptr<DType>());

for (index_t g = 0; g < group_; ++g) {
if (0 == n) {
ASSIGN_DISPATCH(dweight_3d[g], req[conv::kWeight],
Expand Down

0 comments on commit aecdf52

Please sign in to comment.