Searched refs:dropout_3d (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_dropout3d.py | 32 def dropout_3d(keep_prob, nptype): function
73 dropout_3d(0.0, np.float16)
74 dropout_3d(1.0, np.float16)
80 dropout_3d(0.0, np.float32)
81 dropout_3d(1.0, np.float32)
87 dropout_3d(0.0, np.int8)
88 dropout_3d(1.0, np.int8)
94 dropout_3d(0.0, np.int16)
95 dropout_3d(1.0, np.int16)
101 dropout_3d(0.0, np.int32)
[all …]
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 373 self.dropout_3d = nn.Dropout(1 - dropout_rate)
374 self.dropout_3d.dropout.shard(((dp, 1, 1),))
386 output = self.dropout_3d(output)
|