| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_python_pass.py | 46 softmax_model = nn.Softmax()
51 pattern = Call(P.Softmax(), [x])
62 softmax_model = nn.Softmax()
67 sigmoid_softmax_pattern = Prim([P.Sigmoid(), P.Softmax()])
85 softmax_model = nn.Softmax()
90 softmax_pattern = Prim(P.Softmax())
110 softmax_model = nn.Softmax()
115 softmax_pattern = Prim(P.Softmax())
134 softmax_model = nn.Softmax()
139 softmax_pattern = Prim(P.Softmax())
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | softmax.h | 32 class MS_CORE_API Softmax : public PrimitiveC {
35 Softmax() : PrimitiveC(kNameSoftmax) { InitIOName({"x"}, {"output"}); } in Softmax() function
37 ~Softmax() = default;
38 MS_DECLARE_PARENT(Softmax, PrimitiveC);
51 using PrimSoftmaxPtr = std::shared_ptr<Softmax>;
|
| D | softmax.cc | 29 void Softmax::set_axis(const std::vector<int64_t> &axis) { (void)this->AddAttr(kAxis, MakeValue(axi… in set_axis()
31 std::vector<int64_t> Softmax::get_axis() const { in get_axis()
36 void Softmax::Init(const int64_t axis) { in Init()
84 REGISTER_PRIMITIVE_EVAL_IMPL(Softmax, prim::kPrimSoftmax, SoftmaxInfer, nullptr, true);
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_softmax_op.py | 32 self.softmax = P.Softmax(axis=-1)
46 Softmax = NetSoftmax()
47 output = Softmax()
60 self.softmax = P.Softmax(axis=-2)
74 Softmax = NetSoftmax1()
75 output = Softmax()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_softmax_op.py | 30 self.softmax1 = P.Softmax()
31 self.softmax2 = P.Softmax(axis)
50 Softmax = NetSoftmax()
51 output = Softmax(x)
60 Softmax = NetSoftmax()
61 output = Softmax(x)
73 self.softmax1 = P.Softmax()
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_softmax.py | 28 self.Softmax = P.Softmax()
31 return self.Softmax(x)
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_softmax.py | 25 self.Softmax = P.Softmax(axis)
28 return self.Softmax(x)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | softmax_gpu_kernel.cc | 21 MS_REG_GPU_KERNEL_ONE(Softmax, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberT…
23 MS_REG_GPU_KERNEL_ONE(Softmax, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberT…
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | activation_info.h | 108 class Softmax : public ActivationBase {
110 explicit Softmax(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Softmax() function
113 ~Softmax() override = default;
125 class SoftmaxInfo : public Softmax {
129 : Softmax(name, inputs_shape, outputs_shape, attrs) {} in SoftmaxInfo()
133 class LogSoftmaxInfo : public Softmax {
137 : Softmax(name, inputs_shape, outputs_shape, attrs) {} in LogSoftmaxInfo()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | activation_test.cc | 29 class Softmax;
31 using SoftmaxPtr = std::shared_ptr<Softmax>;
71 …soft_ptr_ = std::make_shared<Softmax>("softmax_info", sm_inputs_shape, sm_outputs_shape, softmax_a… in SetUp()
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_softmax.cc | 35 auto softmax = std::make_shared<Softmax>(); in TEST_F()
65 auto softmax = std::make_shared<Softmax>(); in TEST_F()
95 auto softmax = std::make_shared<Softmax>(); in TEST_F()
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_create_obj.py | 43 self.softmax = nn.Softmax(0)
47 x = nn.Softmax(self.axis)(x)
104 self.softmax = nn.Softmax(self.axis)
|
| /third_party/mindspore/tests/ut/python/pynative_mode/nn/ |
| D | test_activation.py | 51 layer = nn.Softmax(1)
60 layer = nn.Softmax()
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | activation.py | 47 class Softmax(Cell): class
91 super(Softmax, self).__init__()
92 self.softmax = P.Softmax(axis)
860 'softmax': Softmax,
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops_reid.py | 79 'block': P.Softmax(),
84 'block': P.Softmax(axis=(0, 1)),
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_hybird_parallel_activation.py | 114 self.softmax = P.Softmax().shard(strategy3)
168 self.softmax = P.Softmax().shard(strategy3)
197 self.softmax = P.Softmax().shard(strategy5)
229 self.softmax = P.Softmax(axis=(0, 1)).shard(strategy5)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | softmax_fp32.h | 25 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | softmax_cpu_kernel.h | 37 MS_REG_CPU_KERNEL(Softmax, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeF…
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_softmax_parser.cc | 25 auto prim = std::make_unique<ops::Softmax>(); in Parse()
|
| /third_party/mindspore/tests/ut/python/exec/ |
| D | test_softmax.py | 27 self.softmax = nn.Softmax(dim)
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_softmax_parser.cc | 25 auto prim = std::make_unique<ops::Softmax>(); in Parse()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | softmax_infer.c | 42 REG_INFER(Softmax, PrimType_Softmax, SoftMaxInferShape)
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_softmax_parser.cc | 31 auto prim = std::make_unique<ops::Softmax>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_softmax_parser.cc | 29 auto prim = std::make_unique<ops::Softmax>(); in Parse()
|
| /third_party/mindspore/tests/st/ops/ascend/ |
| D | test_softmax.py | 29 self.softmax = P.Softmax(axis=1)
|