| 11 | namespace { |
| 12 | |
| 13 | int64_t calc_subsampled_dim(int64_t input_dim, int64_t total_padding, int64_t kernel_size, int layers, int stride) { |
| 14 | int64_t value = input_dim; |
| 15 | for (int i = 0; i < layers; ++i) { |
| 16 | value = (value + total_padding - kernel_size) / stride + 1; |
| 17 | } |
| 18 | return value; |
| 19 | } |
| 20 | |
| 21 | core::TensorValue calc_subsampled_lengths( |
| 22 | core::ModuleBuildContext & ctx, |