Skip to content

GitLab

"examples/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "2ba36b479a8faf0f01d29c5c2991f8bebf3a4efc"
Unverified Commit 59ed0ddd authored by Jaemin Jung's avatar Jaemin Jung Committed by GitHub
Browse files

Fix the dimension (#1117)

parent f7caa80f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/ops/csrc/tensorrt/plugins/trt_modulated_deform_conv.cpp
View file @ 59ed0ddd
...@@ -65,7 +65,7 @@ nvinfer1::DimsExprs ModulatedDeformableConvPluginDynamic::getOutputDimensions( ...@@ -65,7 +65,7 @@ nvinfer1::DimsExprs ModulatedDeformableConvPluginDynamic::getOutputDimensions(
nvinfer1::DimsExprs ret; nvinfer1::DimsExprs ret;
ret.nbDims = 4; ret.nbDims = 4;
ret.d[0] = inputs[0].d[0]; ret.d[0] = inputs[0].d[0];
ret.d[1] = inputs[2].d[0]; ret.d[1] = inputs[3].d[0];
ret.d[2] = inputs[1].d[2]; ret.d[2] = inputs[1].d[2];
ret.d[3] = inputs[1].d[3]; ret.d[3] = inputs[1].d[3];
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment