Skip to content

Commit

Permalink
style(pre-commit): autofix
Browse files Browse the repository at this point in the history
  • Loading branch information
pre-commit-ci[bot] authored and amadeuszsz committed Jan 8, 2025
1 parent a55be36 commit 7ca81ce
Showing 1 changed file with 4 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ class ConvProfiler : public tensorrt_common::Profiler
if (const auto type = layer->getType(); type == nvinfer1::LayerType::kCONVOLUTION) {
const auto name = layer->getName();
auto conv = dynamic_cast<nvinfer1::IConvolutionLayer *>(layer);

nvinfer1::ITensor * in = layer->getInput(0);
nvinfer1::Dims in_dim = in->getDimensions();

Expand All @@ -93,7 +93,9 @@ class ConvProfiler : public tensorrt_common::Profiler
int32_t stride = s_dims.d[0];
int32_t groups = conv->getNbGroups();

layer_dict_.insert_or_assign(name, ConvLayerInfo{in_dim.d[1], out_dim.d[1], in_dim.d[3], in_dim.d[2], kernel, stride, groups, type});
layer_dict_.insert_or_assign(
name, ConvLayerInfo{
in_dim.d[1], out_dim.d[1], in_dim.d[3], in_dim.d[2], kernel, stride, groups, type});
}
}

Expand Down

0 comments on commit 7ca81ce

Please sign in to comment.