| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | dropout_impl.cu | 22 float keep_prob) { in DropoutForwardKernel()
32 size_t num_count, float keep_prob) { in DropoutForwardKernel()
48 float keep_prob) { in DropoutBackwardKernel()
56 float keep_prob) { in DropoutBackwardKernel()
|
| D | dropout3d_impl.cu | 23 … const float keep_prob, const float scale, const size_t num_per_chan) { in Dropout3DForwardKernel()
37 … const size_t num_count, const float keep_prob, const float scale, in Dropout3DForwardKernel()
55 const float keep_prob, const size_t num_per_chan, cudaStream_t cuda_stream) { in Dropout3DForward()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | dropout.cc | 27 void Dropout::Init(const float keep_prob) { this->set_keep_prob(keep_prob); } in Init()
29 void Dropout::set_keep_prob(const float keep_prob) { in set_keep_prob()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_dropout3d.py | 24 def __init__(self, keep_prob): argument
32 def dropout_3d(keep_prob, nptype): argument
|
| D | test_dropout.py | 25 def __init__(self, keep_prob): argument
60 def __init__(self, keep_prob): argument
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | dropout_grad.cc | 27 void DropoutGrad::Init(const float keep_prob) { this->set_keep_prob(keep_prob); } in Init()
29 void DropoutGrad::set_keep_prob(const float keep_prob) { in set_keep_prob()
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_dropout.py | 24 def __init__(self, keep_prob): argument
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_dropout_do_mask.py | 31 def construct(self, x, mask, keep_prob): argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_dropout_grad_op.py | 30 def __init__(self, keep_prob=0.5): argument
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_dropout2d.py | 30 def __init__(self, keep_prob): argument
|
| D | test_dropout3d.py | 30 def __init__(self, keep_prob): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | config.py | 44 keep_prob = 0.9 variable in ModelConfig
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | dropout_grad_kernel.cc | 79 … const std::vector<AddressPtr> &outputs, float keep_prob) { in DropoutBackwardKernel()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_matmul_dropout.py | 61 def __init__(self, keep_prob=0.5, dtype=mstype.float32): argument
|
| /third_party/mindspore/mindspore/nn/probability/toolbox/ |
| D | uncertainty_evaluation.py | 261 def _make_epistemic(self, epi_model, keep_prob=0.5): argument
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 82 keep_prob=0.7, scale_coef=1.0, convert_dtype=True): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | dropout_do_mask_info.cc | 237 ValuePtr keep_prob = GetValueNode(cnode->input(DROPOUT_DO_MASK_KEEP_PROB_INDEX)); in GetDropoutGenMaskReplaceOp() local
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_effect_random.py | 367 def construct(self, keep_prob): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | dropout_unify_mindir.cc | 78 auto keep_prob = AnfAlgo::GetNodeAttr<float>(node, kKeepProb); in CreateKeepPorbValueNode() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | basic.py | 148 def __init__(self, keep_prob=0.5, dtype=mstype.float32): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 7286 def __init__(self, keep_prob=0.5, Seed0=0, Seed1=0): argument
7341 def __init__(self, keep_prob=0.5): argument
7398 def __init__(self, keep_prob=0.5): argument
7580 def __init__(self, keep_prob=1.0, forget_bias=1.0, state_is_tuple=True, activation='tanh'): argument
7721 keep_prob=1.0, argument
7892 keep_prob=1.0, argument
|
| D | _grad_ops.py | 647 def __init__(self, keep_prob=0.5): argument
1375 keep_prob=1.0, argument
1477 keep_prob=1.0, argument
2145 def __init__(self, keep_prob): argument
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_nn_ops.py | 366 def bprop(shape, keep_prob, out, dout): argument
377 def bprop(x, y, keep_prob, out, dout): argument
|