Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!17514 add dropout3d st 

From: @wangshuide2020
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2021-06-02 09:46:26 +08:00 committed by Gitee
parent b21b76de80 b6822c835b
commit 8b942f5666
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/st/ops/gpu/test_dropout3d.py
View File

@ -66,42 +66,42 @@ def dropout_3d(keep_prob, nptype):
assert x_np.shape == output_np.shape == mask_np.shape
@pytest.mark.level2
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_dropout3d_float16():
dropout_3d(0.0, np.float16)
dropout_3d(1.0, np.float16)
@pytest.mark.level2
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_dropout3d_float32():
dropout_3d(0.0, np.float32)
dropout_3d(1.0, np.float32)
@pytest.mark.level2
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_dropout3d_int8():
dropout_3d(0.0, np.int8)
dropout_3d(1.0, np.int8)
@pytest.mark.level2
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_dropout3d_int16():
dropout_3d(0.0, np.int16)
dropout_3d(1.0, np.int16)
@pytest.mark.level2
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_dropout3d_int32():
dropout_3d(0.0, np.int32)
dropout_3d(1.0, np.int32)
@pytest.mark.level2
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_dropout3d_int64():