diff --git a/src/layers/regularizers/distconv/distconv_layer_norm.cu b/src/layers/regularizers/distconv/distconv_layer_norm.cu index ba102f2a897..19788003f68 100644 --- a/src/layers/regularizers/distconv/distconv_layer_norm.cu +++ b/src/layers/regularizers/distconv/distconv_layer_norm.cu @@ -181,6 +181,7 @@ void LayerNormalization::calculate_backward_stats( const DCTensor& statistics, DCTensor& statistics_grad) { + const auto& input_dims = input.get_local_shape(); const auto& statistics_dims = statistics.get_local_shape(); const auto local_num_samples = input_dims[3];