Changeset 8429515 for src/ai/maxpool2d.h

Timestamp:

Dec 29, 2021, 5:52:01 PM (2 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

6627c87

Parents:

514a87c

git-author:

Paul Brossier <piem@piem.org> (01/29/19 02:59:06)

git-committer:

Paul Brossier <piem@piem.org> (12/29/21 17:52:01)

Message:

[maxpool2d] add basic documentation

File:

• src/ai/maxpool2d.h (modified) (1 diff)

src/ai/maxpool2d.h

@@ -22 +22 @@
 #define AUBIO_MAXPOOL2D_H
 
+/** \file
+
+  Max pooling layer (2D)
+
+*/
+
 #ifdef __cplusplus
 extern "C" {
 #endif
 
+/** maxpool2d layer */
 typedef struct _aubio_maxpool2d_t aubio_maxpool2d_t;
 
+/** create a new maxpool2d layer
+
+  \param pool_size  size of the pooling windows
+
+  \return new ::aubio_maxpool2d_t layer
+
+*/
 aubio_maxpool2d_t *new_aubio_maxpool2d(uint_t pool_size[2]);
 
-void aubio_maxpool2d_do(aubio_maxpool2d_t *t,
-        aubio_tensor_t *input_tensor,
-        aubio_tensor_t *activations);
+/** get output shape
 
-uint_t aubio_maxpool2d_set_weights(aubio_maxpool2d_t *t,
-        aubio_tensor_t *kernels);
+  \param t      layer
+  \param input  input tensor
+  \param shape  output shape
 
-aubio_tensor_t *aubio_maxpool2d_get_weigths(aubio_maxpool2d_t *t);
+  \return 0 on success, non-zero otherwise
 
+*/
 uint_t aubio_maxpool2d_get_output_shape(aubio_maxpool2d_t *t,
         aubio_tensor_t *input, uint_t *shape);
 
+/** compute layer output
+
+  \param    t               layer
+  \param    input_tensor    input tensor
+  \param    output_tensor   output tensor
+
+*/
+void aubio_maxpool2d_do(aubio_maxpool2d_t *t,
+        aubio_tensor_t *input_tensor,
+        aubio_tensor_t *output_tensor);
+
+/** destroy layer
+
+  \param t      layer to destroy
+
+*/
 void del_aubio_maxpool2d(aubio_maxpool2d_t *t);