Changeset 217ca89
- Timestamp:
- Jan 29, 2019, 2:59:06 AM (6 years ago)
- Branches:
- feature/crepe_org
- Children:
- b60514b
- Parents:
- 382375d
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
src/ai/maxpool2d.h
r382375d r217ca89 22 22 #define AUBIO_MAXPOOL2D_H 23 23 24 /** \file 25 26 Max pooling layer (2D) 27 28 */ 29 24 30 #ifdef __cplusplus 25 31 extern "C" { 26 32 #endif 27 33 34 /** maxpool2d layer */ 28 35 typedef struct _aubio_maxpool2d_t aubio_maxpool2d_t; 29 36 37 /** create a new maxpool2d layer 38 39 \param pool_size size of the pooling windows 40 41 \return new ::aubio_maxpool2d_t layer 42 43 */ 30 44 aubio_maxpool2d_t *new_aubio_maxpool2d(uint_t pool_size[2]); 31 45 32 void aubio_maxpool2d_do(aubio_maxpool2d_t *t, 33 aubio_tensor_t *input_tensor, 34 aubio_tensor_t *activations); 46 /** get output shape 35 47 36 uint_t aubio_maxpool2d_set_weights(aubio_maxpool2d_t *t, 37 aubio_tensor_t *kernels); 48 \param t layer 49 \param input input tensor 50 \param shape output shape 38 51 39 aubio_tensor_t *aubio_maxpool2d_get_weigths(aubio_maxpool2d_t *t); 52 \return 0 on success, non-zero otherwise 40 53 54 */ 41 55 uint_t aubio_maxpool2d_get_output_shape(aubio_maxpool2d_t *t, 42 56 aubio_tensor_t *input, uint_t *shape); 43 57 58 /** compute layer output 59 60 \param t layer 61 \param input_tensor input tensor 62 \param output_tensor output tensor 63 64 */ 65 void aubio_maxpool2d_do(aubio_maxpool2d_t *t, 66 aubio_tensor_t *input_tensor, 67 aubio_tensor_t *output_tensor); 68 69 /** destroy layer 70 71 \param t layer to destroy 72 73 */ 44 74 void del_aubio_maxpool2d(aubio_maxpool2d_t *t); 45 75
Note: See TracChangeset
for help on using the changeset viewer.