Skip to content

Commit

Permalink
refactor(aten::sum): Apply linting
Browse files Browse the repository at this point in the history
Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>
  • Loading branch information
narendasan committed Feb 8, 2021
1 parent a142f6f commit 3e7cf8e
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions core/conversion/converters/impl/reduce.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -83,13 +83,16 @@ auto reduce_registrations TRTORCH_UNUSED =
c10::List<int64_t> calculated_dims;
auto in_dims = util::toVec(in_tensor->getDimensions());
LOG_DEBUG("InDims " << in_dims); // Some abuse of toDim but just for debug info
LOG_DEBUG("Dim to reduce(original):" << util::toDims(dims)); // Some abuse of toDim but just for debug info
LOG_DEBUG(
"Dim to reduce(original):" << util::toDims(dims)); // Some abuse of toDim but just for debug info
for (int i = 0; i < dims.size(); i++) {
auto dim_val = dims[i] == -1 ? (in_dims.size() - 1) : dims[i];
calculated_dims.push_back(dim_val);
}

LOG_DEBUG("Dim to reduce(converted):" << util::toDims(calculated_dims)); // Some abuse of toDim but just for debug info
LOG_DEBUG(
"Dim to reduce(converted):"
<< util::toDims(calculated_dims)); // Some abuse of toDim but just for debug info

uint32_t axis_mask = 0;
for (size_t d = 0; d < calculated_dims.size(); d++) {
Expand Down

0 comments on commit 3e7cf8e

Please sign in to comment.