| /third_party/mindspore/mindspore/core/ops/ |
| D | sigmoid.h | 27 class MS_CORE_API Sigmoid : public PrimitiveC {
30 Sigmoid() : PrimitiveC(kNameSigmoid) { InitIOName({"x"}, {"output"}); } in Sigmoid() function
32 ~Sigmoid() = default;
33 MS_DECLARE_PARENT(Sigmoid, PrimitiveC);
|
| D | sigmoid.cc | 23 REGISTER_PRIMITIVE_C(kNameSigmoid, Sigmoid);
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nonlinear_fuc_ops_declare.cc | 63 INPUT_MAP(Sigmoid) = {{1, INPUT_DESC(x)}};
64 ATTR_MAP(Sigmoid) = EMPTY_ATTR_MAP;
65 OUTPUT_MAP(Sigmoid) = {{0, OUTPUT_DESC(y)}};
66 REG_ADPT_DESC(Sigmoid, kNameSigmoid, ADPT_DESC(Sigmoid))
|
| D | nonlinear_fuc_ops_declare.h | 89 DECLARE_OP_ADAPTER(Sigmoid)
90 DECLARE_OP_USE_OUTPUT(Sigmoid)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | activation_gpu_kernel.cc | 36 MS_REG_GPU_KERNEL_ONE(Sigmoid, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberT…
38 MS_REG_GPU_KERNEL_ONE(Sigmoid, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberT…
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | activation.py | 484 class Sigmoid(Cell): class
524 super(Sigmoid, self).__init__()
525 self.sigmoid = P.Sigmoid()
868 'sigmoid': Sigmoid,
|
| D | rnns.py | 92 ingate = P.Sigmoid()(ingate)
93 forgetgate = P.Sigmoid()(forgetgate)
95 outgate = P.Sigmoid()(outgate)
114 resetgate = P.Sigmoid()(i_r + h_r)
115 inputgate = P.Sigmoid()(i_i + h_i)
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | glu_fp32.cc | 117 int GluCPUKernel::Sigmoid(int task_id) { in Sigmoid() function in mindspore::kernel::GluCPUKernel
128 return ::Sigmoid(input_addr + stride * task_id, count, output_addr + stride * task_id); in Sigmoid()
154 return activation_kernel->Sigmoid(task_id); in SigmoidRun()
|
| D | glu_fp32.h | 47 int Sigmoid(int task_id);
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | sigmoid.py | 19 class Sigmoid(Expander): class
|
| D | __init__.py | 49 from .sigmoid import Sigmoid
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_sigmoid.py | 28 self.sigmoid = nn.Sigmoid()
|
| D | test_sigmoid_grad.py | 30 self.sigmoid = P.Sigmoid()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | activation_fp32.h | 37 int Sigmoid(const float *src, int length, float *dst);
|
| D | lstm_fp32.c | 205 Sigmoid(input_gate, lstm_param->batch_ * lstm_param->hidden_size_, input_gate); in LstmStepUnit()
208 Sigmoid(forget_gate, lstm_param->batch_ * lstm_param->hidden_size_, forget_gate); in LstmStepUnit()
217 Sigmoid(output_gate, lstm_param->batch_ * lstm_param->hidden_size_, output_gate); in LstmStepUnit()
|
| D | gru_fp32.c | 64 Sigmoid(reset_gate, gru_param->batch_ * gru_param->hidden_size_, reset_gate); in GruStepUnit()
66 Sigmoid(update_gate, gru_param->batch_ * gru_param->hidden_size_, update_gate); in GruStepUnit()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_sigmoid_op.py | 28 self.sigmoid = P.Sigmoid()
|
| /third_party/mindspore/tests/ut/python/nn/probability/dpn/ |
| D | test_vae.py | 41 self.sigmoid = nn.Sigmoid()
|
| /third_party/mindspore/tests/st/probability/dpn/ |
| D | test_gpu_vae_gan.py | 52 self.sigmoid = nn.Sigmoid()
73 self.sigmoid = nn.Sigmoid()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | activation_mapper.cc | 37 {mindspore::SIGMOID, std::make_shared<ops::Sigmoid>()}, in Mapper()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_auto_parallel_activation.py | 28 self.sigmoid = P.Sigmoid().shard(strategy2)
|
| D | test_auto_parallel_shard_propagation2.py | 29 self.sigmoid = P.Sigmoid().shard(strategy3)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | eltwise_cpu_kernel.h | 54 MS_REG_CPU_KERNEL(Sigmoid, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeF…
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | sigmoid_int8_tests.cc | 30 TEST_F(TestSigmoidInt8, Sigmoid) { in TEST_F() argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_sigmoid_op.py | 39 self.ops = P.Sigmoid()
|