There was an error while loading. Please reload this page.
1 parent daef022 commit 5b133d0Copy full SHA for 5b133d0
1 file changed
src/layers/regularizers/distconv/distconv_layer_norm.cu
@@ -179,6 +179,7 @@ void LayerNormalization<Backend, DataType>::calculate_backward_stats(
179
const DCTensor<Allocator>& statistics,
180
DCTensor<Allocator>& statistics_grad)
181
{
182
+
183
const auto& input_dims = input.get_local_shape();
184
const auto& statistics_dims = statistics.get_local_shape();
185
const auto local_num_samples = input_dims[3];
0 commit comments