Changeset be3164d
- Timestamp:
- Dec 29, 2021, 5:52:03 PM (3 years ago)
- Branches:
- feature/crepe
- Children:
- 07c70b4b
- Parents:
- cba2b12
- git-author:
- Paul Brossier <piem@piem.org> (12/27/21 01:38:56)
- git-committer:
- Paul Brossier <piem@piem.org> (12/29/21 17:52:03)
- Location:
- src/ai
- Files:
-
- 6 edited
Legend:
- Unmodified
- Added
- Removed
-
src/ai/batchnorm.c
rcba2b12 rbe3164d 32 32 }; 33 33 34 #if defined(DEBUG) 34 35 static void aubio_batchnorm_debug(aubio_batchnorm_t *c, 35 36 aubio_tensor_t *input_tensor); 37 #endif 36 38 37 39 aubio_batchnorm_t *new_aubio_batchnorm(void) … … 59 61 } 60 62 63 #if defined(DEBUG) 61 64 void aubio_batchnorm_debug(aubio_batchnorm_t *c, aubio_tensor_t *input_tensor) 62 65 { … … 66 69 c->n_outputs, 4 * c->n_outputs); 67 70 } 71 #endif 68 72 69 73 uint_t aubio_batchnorm_get_output_shape(aubio_batchnorm_t *c, … … 93 97 } 94 98 99 #if defined(DEBUG) 95 100 aubio_batchnorm_debug(c, input); 101 #endif 96 102 97 103 return AUBIO_OK; -
src/ai/conv1d.c
rcba2b12 rbe3164d 51 51 }; 52 52 53 #if defined(DEBUG) 53 54 static 54 55 void aubio_conv1d_debug(aubio_conv1d_t *c, aubio_tensor_t *input_tensor); 56 #endif 55 57 56 58 aubio_conv1d_t *new_aubio_conv1d(uint_t n_filters, uint_t kernel_shape[1]) … … 176 178 shape[1] = output_shape[1]; 177 179 180 #if defined(DEBUG) 178 181 aubio_conv1d_debug(c, input_tensor); 182 #endif 179 183 180 184 return AUBIO_OK; 181 185 } 182 186 187 #if defined(DEBUG) 183 188 void aubio_conv1d_debug(aubio_conv1d_t *c, aubio_tensor_t *input_tensor) 184 189 { … … 196 201 -c->padding_start); 197 202 } 203 #endif 198 204 199 205 uint_t aubio_conv1d_check_output_shape(aubio_conv1d_t *c, -
src/ai/conv2d.c
rcba2b12 rbe3164d 50 50 }; 51 51 52 #if defined(DEBUG) 52 53 static void aubio_conv2d_debug(aubio_conv2d_t *c, aubio_tensor_t *input_tensor); 54 #endif 53 55 54 56 aubio_conv2d_t *new_aubio_conv2d(uint_t n_filters, uint_t kernel_shape[2]) … … 203 205 #endif 204 206 207 #if defined(DEBUG) 205 208 aubio_conv2d_debug(c, input_tensor); 209 #endif 206 210 207 211 return AUBIO_OK; 208 212 } 209 213 214 #if defined(DEBUG) 210 215 void aubio_conv2d_debug(aubio_conv2d_t *c, aubio_tensor_t *input_tensor) 211 216 { … … 228 233 -c->padding_start[0], -c->padding_start[1]); 229 234 } 235 #endif 230 236 231 237 uint_t aubio_conv2d_check_output_shape(aubio_conv2d_t *c, -
src/ai/dense.c
rcba2b12 rbe3164d 53 53 } 54 54 55 #if defined(DEBUG) 55 56 void aubio_dense_debug(aubio_dense_t *c, aubio_tensor_t *input_tensor) 56 57 { … … 61 62 c->n_units, c->weights->height, c->weights->length, c->bias->length); 62 63 } 64 #endif 63 65 64 66 uint_t aubio_dense_get_output_shape(aubio_dense_t *c, … … 77 79 if (!c->bias) return AUBIO_FAIL; 78 80 81 #if defined(DEBUG) 79 82 aubio_dense_debug(c, input); 83 #endif 80 84 81 85 return AUBIO_OK; -
src/ai/maxpool1d.c
rcba2b12 rbe3164d 31 31 }; 32 32 33 #if defined(DEBUG) 33 34 static void aubio_maxpool1d_debug(aubio_maxpool1d_t *c, 34 35 aubio_tensor_t *input_tensor); 36 #endif 35 37 36 38 aubio_maxpool1d_t *new_aubio_maxpool1d(uint_t pool_size[1]) … … 56 58 } 57 59 60 #if defined(DEBUG) 58 61 void aubio_maxpool1d_debug(aubio_maxpool1d_t *c, aubio_tensor_t *input_tensor) 59 62 { … … 63 66 input_tensor->shape[1], c->pool_size); 64 67 } 68 #endif 65 69 66 70 uint_t aubio_maxpool1d_get_output_shape(aubio_maxpool1d_t *c, … … 73 77 shape[1] = input->shape[1]; 74 78 79 #if defined(DEBUG) 75 80 aubio_maxpool1d_debug(c, input); 81 #endif 76 82 77 83 return AUBIO_OK; -
src/ai/maxpool2d.c
rcba2b12 rbe3164d 31 31 }; 32 32 33 #if defined(DEBUG) 33 34 static void aubio_maxpool2d_debug(aubio_maxpool2d_t *c, 34 35 aubio_tensor_t *input_tensor); 36 #endif 35 37 36 38 aubio_maxpool2d_t *new_aubio_maxpool2d(uint_t pool_size[2]) … … 59 61 } 60 62 63 #if defined(DEBUG) 61 64 void aubio_maxpool2d_debug(aubio_maxpool2d_t *c, aubio_tensor_t *input_tensor) 62 65 { … … 70 73 c->pool_size[1]); 71 74 } 75 #endif 72 76 73 77 uint_t aubio_maxpool2d_get_output_shape(aubio_maxpool2d_t *c, … … 81 85 shape[2] = input->shape[2]; 82 86 87 #if defined(DEBUG) 83 88 aubio_maxpool2d_debug(c, input); 89 #endif 84 90 85 91 return AUBIO_OK;
Note: See TracChangeset
for help on using the changeset viewer.