// SPDX-License-Identifier: MIT // Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved. #include "gtest/gtest.h" #include "ck/ck.hpp" using F16 = ck::half_t; using F32 = float; using I32 = int32_t; using ck::index_t; struct PoolingParam { PoolingParam(const std::vector& length, const std::vector& window_spatial_lengths, const std::vector& window_strides, const std::vector& input_left_pads, const std::vector& input_right_pads) : length_(length), window_spatial_lengths_(window_spatial_lengths), window_strides_(window_strides), input_left_pads_(input_left_pads), input_right_pads_(input_right_pads) { } std::vector length_; std::vector window_spatial_lengths_; std::vector window_strides_; std::vector input_left_pads_; std::vector input_right_pads_; };