Skip to content

Commit 763952d

Browse files
authored
Support 3D instanceNormalization (#745)
1 parent 2ead004 commit 763952d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: plugin/instanceNormalizationPlugin/instanceNormalizationPlugin.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ int InstanceNormalizationPlugin::enqueue(const nvinfer1::PluginTensorDesc* input
176176
int n = input_dims.d[0];
177177
int c = input_dims.d[1];
178178
int h = input_dims.d[2];
179-
int w = input_dims.d[3];
179+
int w = input_dims.d[3] > 0 ? input_dims.d[3] : 1;
180180
size_t nchan_bytes = c * sizeof(float);
181181

182182
// Note: We repeat the data for each batch entry so that we can do the full

0 commit comments

Comments
 (0)