Skip to content

Commit 28387d9

Browse files
committed
Fix linting
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent a294f9e commit 28387d9

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

core/conversion/converters/impl/pooling.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -80,17 +80,17 @@ bool AdaptivePoolingConverter(
8080
std::copy_n(out_size.d, out_size.nbDims, out_shape.begin() + (in_shape.size() - out_size.nbDims));
8181

8282
std::vector<int32_t> in_shape_casted(in_shape.begin(), in_shape.end());
83-
f.emplace_back(
84-
nvinfer1::PluginField("in_shape", in_shape_casted.data(), nvinfer1::PluginFieldType::kINT32, in_shape.size()));
83+
f.emplace_back(nvinfer1::PluginField(
84+
"in_shape", in_shape_casted.data(), nvinfer1::PluginFieldType::kINT32, in_shape.size()));
8585

8686
std::vector<int32_t> out_shape_casted(out_shape.begin(), out_shape.end());
87-
f.emplace_back(
88-
nvinfer1::PluginField("out_shape", out_shape_casted.data(), nvinfer1::PluginFieldType::kINT32, out_shape.size()));
87+
f.emplace_back(nvinfer1::PluginField(
88+
"out_shape", out_shape_casted.data(), nvinfer1::PluginFieldType::kINT32, out_shape.size()));
8989

9090
auto out_size_vec = util::toVec(out_size);
9191
std::vector<int32_t> out_size_casted(out_size_vec.begin(), out_size_vec.end());
92-
f.emplace_back(
93-
nvinfer1::PluginField("out_size", out_size_casted.data(), nvinfer1::PluginFieldType::kINT32, out_size_vec.size()));
92+
f.emplace_back(nvinfer1::PluginField(
93+
"out_size", out_size_casted.data(), nvinfer1::PluginFieldType::kINT32, out_size_vec.size()));
9494

9595
f.emplace_back(nvinfer1::PluginField("scales", nullptr, nvinfer1::PluginFieldType::kFLOAT64, 0));
9696

core/plugins/impl/normalize_plugin.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ nvinfer1::DimsExprs NormalizePlugin::getOutputDimensions(
7272
// For order-0 norm, when the norm dimension is None, it should normalize across all dimensions.
7373
// TODO: For dim=None, the axes_ passed would have [0, 0, 0] which is obtained through loop counter in TRTorch.
7474
// Resolve this. For dim=None case, change the axes_ inplace to range(0, axes_.size())
75-
bool isAxisNone =
76-
std::all_of(axes_.begin(), axes_.end(), [](int32_t i) { return i == 0; }) && ((int32_t) axes_.size() == inputs[0].nbDims);
75+
bool isAxisNone = std::all_of(axes_.begin(), axes_.end(), [](int32_t i) { return i == 0; }) &&
76+
((int32_t) axes_.size() == inputs[0].nbDims);
7777
if (isAxisNone) {
7878
std::iota(axes_.data(), axes_.data() + axes_.size(), 0);
7979
}

0 commit comments

Comments
 (0)