Home
last modified time | relevance | path

Searched defs:axis_offset (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dsoftmax.c45 int axis_offset = inner_offset + j; in SoftMaxP1Simple() local
51 int axis_offset = inner_offset + j; in SoftMaxP1Simple() local
71 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local
75 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local
80 int axis_offset = inner_offset + j * inner_size; in SoftMaxP1() local
94 int axis_offset = outter_offset + j * inner_size; in SoftMaxP2() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dsoftmax_int8.c41 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8() local
47 int axis_offset = outter_offset + c + i * inner_size; in SoftmaxInt8() local
58 int axis_offset = outter_offset + i * inner_size; in SoftmaxInt8() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dlog_softmax_fp32.c67 int axis_offset = inner_offset + j * inner_size; in LogSoftmax() local
71 int axis_offset = inner_offset + j * inner_size; in LogSoftmax() local
81 int axis_offset = outter_offset + j * inner_size; in LogSoftmax() local
Dsoftmax_fp32.c126 int axis_offset = inner_offset + j * inner_size; in Softmax() local
130 int axis_offset = inner_offset + j * inner_size; in Softmax() local
140 int axis_offset = outter_offset + j * inner_size; in Softmax() local
Dscale_fp32.c26 int axis_offset = out_offset + i * inner_size; in ScaleInner() local
116 int axis_offset = out_offset + i * inner_size; in ScaleInnerRelu() local
220 int axis_offset = out_offset + i * inner_size; in ScaleInnerRelu6() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dlog_softmax_fp16.c70 int axis_offset = inner_offset + j * inner_size; in LogSoftmaxFp16() local
74 int axis_offset = inner_offset + j * inner_size; in LogSoftmaxFp16() local
84 int axis_offset = outter_offset + j * inner_size; in LogSoftmaxFp16() local
Dsoftmax_fp16.c116 int axis_offset = inner_offset + j * inner_size; in SoftmaxFp16() local
120 int axis_offset = inner_offset + j * inner_size; in SoftmaxFp16() local
130 int axis_offset = outter_offset + j * inner_size; in SoftmaxFp16() local
Dscale_fp16.c24 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInner() local
90 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInnerRelu() local
163 int axis_offset = out_offset + i * inner_size; in Fp16ScaleInnerRelu6() local
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/operator_library/kernels/nnacl/fp32/
Dsoftmax_fp32.c125 int axis_offset = inner_offset + j * inner_size; in Softmax() local
129 int axis_offset = inner_offset + j * inner_size; in Softmax() local
139 int axis_offset = outter_offset + j * inner_size; in Softmax() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconcat_base.c30 int axis_offset = 0; in Concat() local