Skip to content

Commit

Permalink
Fix indent
Browse files Browse the repository at this point in the history
  • Loading branch information
vinx13 committed Nov 22, 2018
1 parent 9aafe00 commit 4302789
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions src/relay/pass/combine_parallel_conv2d.cc
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,7 @@ class ParallelConv2DCombiner {
size_t arg_channel_pos = channel_pos - toutput_a->shape.size() + ta->shape.size();

// Channel super-dimension shoule be present and not broadcasted
if ((arg_channel_pos > channel_pos) || // size_t overflow
if ((arg_channel_pos > channel_pos) || // size_t overflow
!eq(ta->shape[arg_channel_pos], toutput_a->shape[channel_pos]) ||
!eq(tb->shape[arg_channel_pos], toutput_b->shape[channel_pos]))
return false;
Expand Down Expand Up @@ -270,18 +270,18 @@ class ParallelConv2DCombiner {
size_t channel_pos) {
int64_t index = 0;
for (const auto& branch : branches) {
const CallNode* conv2d = branch[0];
int64_t channels = GetConv2DSuperChannelsDim(conv2d);
Array<Integer> begin;
Array<Integer> end;
for (size_t i = 0; i < channel_pos; i++) {
begin.push_back(0);
end.push_back(NullValue<Integer>());
}
begin.push_back(index);
index += channels;
end.push_back(index);
auto slice = MakeStridedSlice(data, std::move(begin), std::move(end), Array<Integer>{});
const CallNode* conv2d = branch[0];
int64_t channels = GetConv2DSuperChannelsDim(conv2d);
Array<Integer> begin;
Array<Integer> end;
for (size_t i = 0; i < channel_pos; i++) {
begin.push_back(0);
end.push_back(NullValue<Integer>());
}
begin.push_back(index);
index += channels;
end.push_back(index);
auto slice = MakeStridedSlice(data, std::move(begin), std::move(end), Array<Integer>{});
subst_map_[GetRef<Expr>(branch[depth])] = slice;
}
}
Expand Down

0 comments on commit 4302789

Please sign in to comment.