Searched refs:SoftShrink (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | soft_shrink.h | 31 class MS_CORE_API SoftShrink : public PrimitiveC {
34 SoftShrink() : PrimitiveC(kNameSoftShrink) { InitIOName({"input_x"}, {"output"}); } in SoftShrink() function
36 ~SoftShrink() = default;
37 MS_DECLARE_PARENT(SoftShrink, PrimitiveC);
43 using PrimSoftShrinkPtr = std::shared_ptr<SoftShrink>;
|
| D | soft_shrink.cc | 61 REGISTER_PRIMITIVE_EVAL_IMPL(SoftShrink, prim::kPrimSoftShrink, SoftShrinkInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | activation.py | 765 class SoftShrink(Cell): class
806 super(SoftShrink, self).__init__()
807 self.softshrink = P.SoftShrink(lambd)
874 'softshrink': SoftShrink,
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_nn_ops.py | 51 @bprop_getters.register(P.SoftShrink)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 89 … ApplyRMSProp, ApplyCenteredRMSProp, BasicLSTMCell, InTopK, AdaptiveAvgPool2D, SoftShrink)
|
| D | nn_ops.py | 8929 class SoftShrink(Primitive): class
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 2266 'block': P.SoftShrink(),
|