| 27 | namespace window_util { |
| 28 | |
| 29 | Window MakeWindow(absl::Span<const int64> sizes) { |
| 30 | Window window; |
| 31 | for (int64 size : sizes) { |
| 32 | auto* dimension = window.add_dimensions(); |
| 33 | dimension->set_size(size); |
| 34 | dimension->set_stride(1); |
| 35 | dimension->set_base_dilation(1); |
| 36 | dimension->set_window_dilation(1); |
| 37 | } |
| 38 | return window; |
| 39 | } |
| 40 | |
| 41 | Window MakeWindow(absl::Span<const int64> sizes, |
| 42 | absl::Span<const int64> strides) { |