@@ -43,8 +43,10 @@ class OpMaxPool2DWithIndicesOutTest : public OperatorTest {
4343};
4444
4545TEST_F (OpMaxPool2DWithIndicesOutTest, SanityTest4D) {
46- torch::executor::testing::TensorFactory<executorch::aten::ScalarType::Float> tfFloat;
47- torch::executor::testing::TensorFactory<executorch::aten::ScalarType::Long> tfLong;
46+ torch::executor::testing::TensorFactory<executorch::aten::ScalarType::Float>
47+ tfFloat;
48+ torch::executor::testing::TensorFactory<executorch::aten::ScalarType::Long>
49+ tfLong;
4850
4951 executorch::aten::Tensor self = tfFloat.make (
5052 {2 , 3 , 5 , 5 },
@@ -68,17 +70,20 @@ TEST_F(OpMaxPool2DWithIndicesOutTest, SanityTest4D) {
6870 26.75 , 68.25 , -24.625 , -53.0 , 51.0 , 90.625 , 65.375 , 43.875 ,
6971 90.875 , -41.625 , 99.875 , 6.375 , -31.25 , -94.0 });
7072 ::std::vector<int64_t > kernel_size_vec = {2 , 2 };
71- executorch::aten::ArrayRef<int64_t > kernel_size = executorch::aten::ArrayRef<int64_t >(
72- kernel_size_vec.data (), kernel_size_vec.size ());
73+ executorch::aten::ArrayRef<int64_t > kernel_size =
74+ executorch::aten::ArrayRef<int64_t >(
75+ kernel_size_vec.data (), kernel_size_vec.size ());
7376 ::std::vector<int64_t > stride_vec = {1 , 1 };
7477 executorch::aten::ArrayRef<int64_t > stride =
7578 executorch::aten::ArrayRef<int64_t >(stride_vec.data (), stride_vec.size ());
7679 ::std::vector<int64_t > padding_vec = {0 , 0 };
7780 executorch::aten::ArrayRef<int64_t > padding =
78- executorch::aten::ArrayRef<int64_t >(padding_vec.data (), padding_vec.size ());
81+ executorch::aten::ArrayRef<int64_t >(
82+ padding_vec.data (), padding_vec.size ());
7983 ::std::vector<int64_t > dilation_vec = {1 , 1 };
8084 executorch::aten::ArrayRef<int64_t > dilation =
81- executorch::aten::ArrayRef<int64_t >(dilation_vec.data (), dilation_vec.size ());
85+ executorch::aten::ArrayRef<int64_t >(
86+ dilation_vec.data (), dilation_vec.size ());
8287 bool ceil_mode = false ;
8388 executorch::aten::Tensor out = tfFloat.zeros ({2 , 3 , 4 , 4 });
8489 executorch::aten::Tensor indices = tfLong.zeros ({2 , 3 , 4 , 4 });
0 commit comments