Home
last modified time | relevance | path

Searched defs:Sigmoid (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dsigmoid.h30 Sigmoid() : PrimitiveC(kNameSigmoid) { InitIOName({"x"}, {"output"}); } in Sigmoid() function
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dsigmoid.py19 class Sigmoid(Expander): class
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dsigmoid_int8_tests.cc30 TEST_F(TestSigmoidInt8, Sigmoid) { in TEST_F() argument
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dglu_fp32.cc117 int GluCPUKernel::Sigmoid(int task_id) { in Sigmoid() function in mindspore::kernel::GluCPUKernel
/third_party/mindspore/mindspore/nn/layer/
Dactivation.py484 class Sigmoid(Cell): class
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dactivation_fp32.c103 int Sigmoid(const float *src, int length, float *dst) { in Sigmoid() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_op_converter.cc345 MS_TRT_CONVERTER_FUNC_REG(Sigmoid) { return AddActivationLayer(node, context, nvinfer1::ActivationT… in MS_TRT_CONVERTER_FUNC_REG() argument
/third_party/mindspore/mindspore/ops/operations/
Dnn_ops.py758 class Sigmoid(PrimitiveWithInfer): class