Searched defs:Softmax (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | softmax.h | 35 Softmax() : PrimitiveC(kNameSoftmax) { InitIOName({"x"}, {"output"}); } in Softmax() function
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | softmax.py | 24 class Softmax(Expander): class
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | softmax_fp32.c | 105 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa… in Softmax() function
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/operator_library/kernels/nnacl/fp32/ |
| D | softmax_fp32.c | 105 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa… in Softmax() function
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | activation.py | 47 class Softmax(Cell): class
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | activation_info.h | 110 explicit Softmax(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Softmax() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 795 MS_TRT_CONVERTER_FUNC_REG(Softmax) { in MS_TRT_CONVERTER_FUNC_REG() argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 239 class Softmax(Primitive): class
|