Commit 21ee3fc0 authored by charlie's avatar charlie
Browse files

pad_calc works with 1D convolutions

parent 63ada580
......@@ -59,7 +59,7 @@ std::vector<std::size_t> calc_dyn_auto_pad(const std::vector<std::size_t>& input
bool use_upper)
{
std::vector<std::size_t> padding;
assert(input_lens.size() >= 4);
assert(input_lens.size() >= 3);
std::size_t num_spatial_dims = input_lens.size() - 2;
padding.resize(2 * num_spatial_dims);
for(std::size_t i = 0; i < num_spatial_dims; i++)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment