| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | hshrink_impl.cu | 21 __global__ void HShrinkKernel(size_t size, const T *input, const float lambd, T *output) { in HShrinkKernel()
31 __global__ void HShrinkGradKernel(size_t size, const T *dout, const T *x, const float lambd, T *out… in HShrinkGradKernel()
41 cudaError_t CalHShrink(const size_t &size, const T *input, const float lambd, T *output, const uint… in CalHShrink()
49 cudaError_t CalHShrinkGrad(const size_t &size, const T *dout, const T *x, const float lambd, T *out… in CalHShrinkGrad()
|
| D | soft_shrink_impl.cu | 21 __global__ void SoftShrinkComp(size_t size, const T *input, const float lambd, T *output) { in SoftShrinkComp()
33 … SoftShrinkGradComp(size_t size, const T *dy_addr, const T *x_addr, const float lambd, T *dx_addr)… in SoftShrinkGradComp()
43 cudaError_t SoftShrink(const size_t &size, const T *input, const float lambd, T *output, const uint… in SoftShrink()
50 cudaError_t SoftShrinkGrad(const size_t &size, const T *dy_addr, const T *x_addr, const float lambd… in SoftShrinkGrad()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_soft_shrink_op.py | 27 def soft_shrink_op_np_bencmark(input_x, lambd): argument
45 def __init__(self, lambd): argument
59 def test_soft_shrink(dtype, data_shape, lambd): argument
135 def softshrink_grad_op_np_bencmark(grad, input_x, lambd=0.5): argument
|
| D | test_hshrink_grad_op.py | 30 def __init__(self, lambd=0.5): argument
38 def hshrink_grad_op_np_bencmark(grad, input_x, lambd): argument
59 def test_hshrink_grad(dtype, data_shape, lambd): argument
|
| D | test_hshrink_op.py | 26 def hshrink_op_np_bencmark(input_x, lambd): argument
47 def test_hshrink(dtype, data_shape, lambd): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/ |
| D | test_hshrink.py | 26 def hshrink_forward_func(input_x, lambd): argument
31 def hshrink_backward_func(input_x, lambd): argument
36 def hshrink_dyn_shape_func(input_x, lambd): argument
|
| D | test_hshrink_grad.py | 26 def hshrink_grad_forward_func(gradients, features, lambd): argument
31 def hshrink_grad_dyn_shape_func(gradients, features, lambd): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_hshrink_op.py | 25 def hshrink_op_np_bencmark(input_x, lambd): argument
46 def test_hshrink(dtype, data_shape, lambd): argument
|
| D | test_soft_shrink_op.py | 30 def soft_shrink_op_np_bencmark(input_x, lambd): argument
48 def __init__(self, lambd): argument
62 def test_soft_shrink(dtype, data_shape, lambd): argument
|
| D | test_hshrink_grad_op.py | 25 def hshrink_grad_op_np_bencmark(grad, input_x, lambd): argument
46 def test_hshrink_grad(dtype, data_shape, lambd): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/experimental/optim/ |
| D | asgd.py | 35 def _run_asgd_opt(lambd, alpha, t0, step, lr, param, grad, eta, mu, ax): argument
103 …def __init__(self, params, lr=1e-2, lambd=1e-4, alpha=0.75, t0=1e6, weight_decay=0.0, maximize=Fal… argument
128 def implementation(self, lambd, alpha, t0, lr, group_id, maximize, gradients, weight_decay): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_hshrink.cc | 33 ValuePtr lambd; member
52 auto lambd = shape_param.lambd->ToAbstract(); in TEST_P() local
|
| D | test_ops_hshrink_grad.cc | 34 ValuePtr lambd; member
59 auto lambd = shape_param.lambd->ToAbstract(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | hshrink_cpu_kernel.h | 44 float lambd = 0.5; variable
|
| D | hshrink_grad_cpu_kernel.h | 44 float lambd = 0.5; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | soft_shrink.cc | 43 void SoftShrink::Init(const float &lambd) { in Init()
48 void SoftShrink::set_lambd(const float &lambd) { in set_lambd()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | soft_shrink_grad.cc | 45 void SoftShrinkGrad::Init(const float &lambd) { in Init()
50 void SoftShrinkGrad::set_lambd(const float &lambd) { in set_lambd()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_shrink_parser.cc | 31 float lambd = 0.5; in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/ |
| D | hshrink_modify_lambd.cc | 30 auto lambd = GetValue<float>(lambd_value_node->value()); in HShrinkModifyLambd() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | hshrink_gpu_kernel.h | 51 float lambd = 0.f; variable
|
| D | hshrink_grad_gpu_kernel.h | 51 float lambd = 0.f; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/onnx/ |
| D | onnx_shrink_parser.cc | 36 float lambd = 0.5; in Parse() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/ |
| D | shrink_mapper.cc | 47 auto lambd = src_prim->GetAttr(mindspore::ops::kLambd); in Mapper() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ |
| D | asgd.py | 156 def __init__(self, params, learning_rate=0.1, lambd=1e-4, alpha=0.75, t0=1e6, weight_decay=0.0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | activation_grad_neon.h | 32 int length, float *dst, float lambd) { in ShrinkGradNEON()
|