| 1396 | } |
| 1397 | |
| 1398 | static std::size_t get_len(const dimension::sub& s, const std::vector<std::size_t>& input_dims) |
| 1399 | { |
| 1400 | if(input_dims.empty()) |
| 1401 | return s.len; |
| 1402 | if(s.axis.empty()) |
| 1403 | return s.len; |
| 1404 | auto dim = input_dims.at(s.axis.front()); |
| 1405 | if(dim == 0) |
| 1406 | return s.len; |
| 1407 | if(dim == 1) |
| 1408 | return 1; |
| 1409 | if(s.axis.size() == 1) |
| 1410 | return dim; |
| 1411 | return s.len; |
| 1412 | } |
| 1413 | |
| 1414 | static operation make_reshape_squeeze(const std::vector<dimension>& new_dims) |
| 1415 | { |
no test coverage detected