Home
last modified time | relevance | path

Searched refs:Softplus (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/tests/ut/python/nn/probability/bijector/
Dtest_softplus.py26 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/
Dsoftplus.h31 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);
Dsoftplus.cc54 REGISTER_PRIMITIVE_EVAL_IMPL(Softplus, prim::kPrimSoftplus, SoftplusInfer, nullptr, true);
/third_party/mindspore/tests/st/probability/bijector/
Dtest_softplus.py31 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/
Dsoftplus_impl.cu37 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…
Dsoftplus_impl.cuh22 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/
Dnonlinear_fuc_ops_declare.cc93 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))
Dnonlinear_fuc_ops_declare.h41 DECLARE_OP_ADAPTER(Softplus)
42 DECLARE_OP_USE_OUTPUT(Softplus)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dsoftplus_gpu_kernel.cc21 MS_REG_GPU_KERNEL_ONE(Softplus, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumber…
23 MS_REG_GPU_KERNEL_ONE(Softplus, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumber…
Dsoftplus_gpu_kernel.h45Softplus(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__.py23 from .softplus import Softplus
Dsoftplus.py23 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/
Dactivation_fp32.h44 int Softplus(const float *src, int length, float *dst);
Dactivation_fp32.c286 int Softplus(const float *src, int length, float *dst) { in Softplus() function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Deltwise_cpu_kernel.h60 MS_REG_CPU_KERNEL(Softplus, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberType…
/third_party/mindspore/tests/st/ops/gpu/
Dtest_softplus_grad_op.py31 self.softplus = P.Softplus()
Dtest_softplus_op.py30 self.softplus = P.Softplus()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_softplus_grad_op.py31 self.softplus = P.Softplus()
Dtest_softplus_op.py30 self.softplus = P.Softplus()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dactivation_fp32.cc92 ret = Softplus(input_addr + stride * task_id, count, output_addr + stride * task_id); in DoActivation()
Daffine_fp32.cc57 return Softplus(data, length, data); in DoActivation()
/third_party/mindspore/mindspore/ops/operations/
D__init__.py81 …SmoothL1Loss, SoftMarginLoss, Softmax, Softsign, Softplus, LRN, RNNTLoss, DynamicRNN, DynamicGRUV2,
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dactivation_fp32_test.cc238 TEST_F(TestActivationFp32, Softplus) { in TEST_F() argument
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_nn_ops.py388 softplus = P.Softplus()
596 @bprop_getters.register(P.Softplus)
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc284 OPERATOR_ONNX_CONVERT_DEFINE(Softplus, Softplus, OpNameInfo())
329 fn(OP_CONVERT_FUNCTION_NAME(Softplus)());

12