@@ -138,7 +138,7 @@ def max_pool2D_forward_naive(
138138 return out , mask
139139
140140
141- def max_pool3d_with_index_wapper (
141+ def max_pool3d_with_index_wrapper (
142142 x ,
143143 kernel_size = [],
144144 strides = [],
@@ -214,7 +214,7 @@ def test_check_grad(self):
214214
215215 def init_test_case (self ):
216216 self .op_type = "max_pool3d_with_index"
217- self .python_api = max_pool3d_with_index_wapper
217+ self .python_api = max_pool3d_with_index_wrapper
218218 self .pool_forward_naive = max_pool3D_forward_naive
219219 self .shape = [2 , 3 , 7 , 7 , 7 ]
220220 self .ksize = [3 , 3 , 3 ]
@@ -236,7 +236,7 @@ def init_global(self):
236236class TestCase2 (TestMaxPoolWithIndex_Op ):
237237 def init_test_case (self ):
238238 self .op_type = "max_pool3d_with_index"
239- self .python_api = max_pool3d_with_index_wapper
239+ self .python_api = max_pool3d_with_index_wrapper
240240 self .pool_forward_naive = max_pool3D_forward_naive
241241 self .shape = [2 , 3 , 7 , 7 , 7 ]
242242 self .ksize = [3 , 3 , 3 ]
@@ -339,7 +339,7 @@ def test_check_grad(self):
339339
340340
341341# ----------------max_pool2d_with_index----------------
342- def max_pool2d_with_index_wapper (
342+ def max_pool2d_with_index_wrapper (
343343 x ,
344344 kernel_size = [],
345345 strides = [],
@@ -356,7 +356,7 @@ def max_pool2d_with_index_wapper(
356356class TestCase4 (TestMaxPoolWithIndex_Op ):
357357 def init_test_case (self ):
358358 self .op_type = "max_pool2d_with_index"
359- self .python_api = max_pool2d_with_index_wapper
359+ self .python_api = max_pool2d_with_index_wrapper
360360 self .pool_forward_naive = max_pool2D_forward_naive
361361 self .shape = [2 , 3 , 7 , 7 ]
362362 self .ksize = [3 , 3 ]
@@ -375,7 +375,7 @@ def init_global(self):
375375class TestCase6 (TestMaxPoolWithIndex_Op ):
376376 def init_test_case (self ):
377377 self .op_type = "max_pool2d_with_index"
378- self .python_api = max_pool2d_with_index_wapper
378+ self .python_api = max_pool2d_with_index_wrapper
379379 self .pool_forward_naive = max_pool2D_forward_naive
380380 self .shape = [2 , 3 , 7 , 7 ]
381381 self .ksize = [3 , 3 ]
0 commit comments