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

add dropout3d st.

This commit is contained in:
wangshuide2020 2021-06-01 19:12:37 +08:00
parent 24c53efb49
commit b6822c835b
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():