Skip to content

Commit

Permalink
fix format
Browse files Browse the repository at this point in the history
  • Loading branch information
Seanlinx committed May 25, 2016
1 parent 119c290 commit 5282930
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/operator/assign_sample-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ class AssignSampleOp : public Operator {
Tensor<xpu, 4> regmask = ctx.requested[assignsample::kTempSpace].get_space<xpu>(s4, s);
Tensor<cpu, 4> regmask_cpu = NewTensor<cpu, real_t>(s4, 0.0f);
for (int i = 0; i < param_.anchor_num; ++i) {
for (int j = i*4; j < (i+1)*4; ++j) {
for (int j = i * 4; j < (i + 1) * 4; ++j) {
regmask_cpu[0][j] += mask_reshape[0][i];
}
}
Expand Down Expand Up @@ -168,7 +168,9 @@ class AssignSampleOp : public Operator {
Copy(maskout_cpu, maskout, s);
int p_count = 0;
for (int i = 0; i < maskout_cpu.size(1); ++i) {
if (maskout_cpu[0][i] == 1) { p_count += 1; }
if (maskout_cpu[0][i] == 1) {
p_count += 1;
}
}

Shape<2> dshape2 = Shape2(2, out_data[assignsample::kMaskout].Size());
Expand Down

0 comments on commit 5282930

Please sign in to comment.