/third_party/mindspore/tests/ut/python/nn/probability/bijector/ |
D | test_softplus.py | 26 b = msb.Softplus() 28 b = msb.Softplus(1.0) 33 msb.Softplus(sharpness='sharpness') 35 msb.Softplus(name=0.1) 43 self.b1 = msb.Softplus(2.0) 44 self.b2 = msb.Softplus() 66 self.b1 = msb.Softplus(2.0) 67 self.b2 = msb.Softplus() 90 self.b1 = msb.Softplus(2.0) 91 self.b2 = msb.Softplus() [all …]
|
/third_party/mindspore/mindspore/core/ops/ |
D | softplus.h | 31 class MS_CORE_API Softplus : public PrimitiveC { 34 Softplus() : PrimitiveC(prim::kPrimSoftplus->name()) { InitIOName({"x"}, {"output"}); } in Softplus() function 36 ~Softplus() = default; 37 MS_DECLARE_PARENT(Softplus, PrimitiveC);
|
D | softplus.cc | 54 REGISTER_PRIMITIVE_EVAL_IMPL(Softplus, prim::kPrimSoftplus, SoftplusInfer, nullptr, true);
|
/third_party/mindspore/tests/st/probability/bijector/ |
D | test_softplus.py | 31 self.bijector = msb.Softplus(sharpness=2.0) 50 self.bijector = msb.Softplus(sharpness=2.0) 69 self.bijector = msb.Softplus(sharpness=2.0) 88 self.bijector = msb.Softplus(sharpness=2.0)
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
D | softplus_impl.cu | 37 void Softplus(const size_t size, const T *input_addr, T *output_addr, cudaStream_t cuda_stream) { in Softplus() function 42 void Softplus(const size_t size, const half *input_addr, half *output_addr, cudaStream_t cuda_strea… in Softplus() function 74 template void Softplus(const size_t size, const float *input_addr, float *output_addr, cudaStream_t… 75 template void Softplus(const size_t size, const half *input_addr, half *output_addr, cudaStream_t c…
|
D | softplus_impl.cuh | 22 void Softplus(const size_t input_size, const T* input_addr, T* output_addr, cudaStream_t cuda_strea…
|
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
D | nonlinear_fuc_ops_declare.cc | 93 INPUT_MAP(Softplus) = {{1, INPUT_DESC(x)}}; 94 ATTR_MAP(Softplus) = EMPTY_ATTR_MAP; 95 OUTPUT_MAP(Softplus) = {{0, OUTPUT_DESC(y)}}; 96 REG_ADPT_DESC(Softplus, kNameSoftplus, ADPT_DESC(Softplus))
|
D | nonlinear_fuc_ops_declare.h | 41 DECLARE_OP_ADAPTER(Softplus) 42 DECLARE_OP_USE_OUTPUT(Softplus)
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
D | softplus_gpu_kernel.cc | 21 MS_REG_GPU_KERNEL_ONE(Softplus, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumber… 23 MS_REG_GPU_KERNEL_ONE(Softplus, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumber…
|
D | softplus_gpu_kernel.h | 45 …Softplus(input_size_ / sizeof(T), input_addr, output_addr, reinterpret_cast<cudaStream_t>(stream_p… in Launch()
|
/third_party/mindspore/mindspore/nn/probability/bijector/ |
D | __init__.py | 23 from .softplus import Softplus
|
D | softplus.py | 23 class Softplus(Bijector): class 78 super(Softplus, self).__init__(name=name, dtype=None, param=param)
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
D | activation_fp32.h | 44 int Softplus(const float *src, int length, float *dst);
|
D | activation_fp32.c | 286 int Softplus(const float *src, int length, float *dst) { in Softplus() function
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
D | eltwise_cpu_kernel.h | 60 MS_REG_CPU_KERNEL(Softplus, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberType…
|
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_softplus_grad_op.py | 31 self.softplus = P.Softplus()
|
D | test_softplus_op.py | 30 self.softplus = P.Softplus()
|
/third_party/mindspore/tests/st/ops/cpu/ |
D | test_softplus_grad_op.py | 31 self.softplus = P.Softplus()
|
D | test_softplus_op.py | 30 self.softplus = P.Softplus()
|
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
D | activation_fp32.cc | 92 ret = Softplus(input_addr + stride * task_id, count, output_addr + stride * task_id); in DoActivation()
|
D | affine_fp32.cc | 57 return Softplus(data, length, data); in DoActivation()
|
/third_party/mindspore/mindspore/ops/operations/ |
D | __init__.py | 81 …SmoothL1Loss, SoftMarginLoss, Softmax, Softsign, Softplus, LRN, RNNTLoss, DynamicRNN, DynamicGRUV2,
|
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
D | activation_fp32_test.cc | 238 TEST_F(TestActivationFp32, Softplus) { in TEST_F() argument
|
/third_party/mindspore/mindspore/ops/_grad/ |
D | grad_nn_ops.py | 388 softplus = P.Softplus() 596 @bprop_getters.register(P.Softplus)
|
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
D | onnx_exporter.cc | 284 OPERATOR_ONNX_CONVERT_DEFINE(Softplus, Softplus, OpNameInfo()) 329 fn(OP_CONVERT_FUNCTION_NAME(Softplus)());
|