From 7897b518befe96aec798d36a5ac4326adf464767 Mon Sep 17 00:00:00 2001 From: Paul Brossier Date: Tue, 29 Jan 2019 01:08:00 +0100 Subject: [PATCH] [batchnorm] improve docs for return values --- src/ai/batchnorm.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ai/batchnorm.h b/src/ai/batchnorm.h index 8d389a54..b299fedb 100644 --- a/src/ai/batchnorm.h +++ b/src/ai/batchnorm.h @@ -56,6 +56,8 @@ aubio_batchnorm_t *new_aubio_batchnorm(void); \param input input tensor \param shape output shape + \return 0 on success, non-zero otherwise + This function determines the number of output channels required and allocate the vectors of weights. The ouptut shape of this layer is identical to the input shape. @@ -159,8 +161,6 @@ uint_t aubio_batchnorm_set_moving_variance(aubio_batchnorm_t *t, \param input_tensor input tensor \param activations output tensor - \return 0 on success, non-zero otherwise. - */ void aubio_batchnorm_do(aubio_batchnorm_t *t, aubio_tensor_t *input_tensor, aubio_tensor_t *activations); -- 2.11.0