fix avg_pool3d count_include_pad as True,test=develop (#27155)

ut_timeout_modifed
Double_V 4 years ago committed by GitHub
parent eb01976037
commit c7b9d97fa9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -165,7 +165,6 @@ class TestPool3d_API(unittest.TestCase):
self.assertTrue(np.allclose(result.numpy(), result_np))
def check_max_dygraph_ndhwc_results(self, place):
print("run ndchw max pool3d")
with fluid.dygraph.guard(place):
input_np = np.random.random([2, 3, 32, 32, 32]).astype("float32")
input = fluid.dygraph.to_variable(
@ -190,7 +189,6 @@ class TestPool3d_API(unittest.TestCase):
np.transpose(result.numpy(), [0, 4, 1, 2, 3]), result_np))
def check_max_dygraph_ceilmode_results(self, place):
print("run ceil mode max pool3d")
with fluid.dygraph.guard(place):
input_np = np.random.random([2, 3, 32, 32, 32]).astype("float32")
input = fluid.dygraph.to_variable(input_np)

@ -389,7 +389,7 @@ def avg_pool3d(x,
stride=None,
padding=0,
ceil_mode=False,
count_include_pad=False,
count_include_pad=True,
divisor_override=None,
data_format="NCDHW",
name=None):

Loading…
Cancel
Save