Home
last modified time | relevance | path

Searched refs:sum_data (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/
Dsoftmax_grad_fp32_tests.cc66 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
67 ASSERT_NE(sum_data, nullptr); in TEST_F()
86 SoftmaxGrad(input_data, yt_data, out_data, sum_data, sum_mul, softmax_param); in TEST_F()
92 SoftmaxGrad(input_data, yt_data, out_data, sum_data, sum_mul, softmax_param); in TEST_F()
107 delete[] sum_data; in TEST_F()
126 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
127 ASSERT_NE(sum_data, nullptr); in TEST_F()
150 SoftmaxGrad(input_data, yt_data, out_data, sum_data, sum_mul, softmax_param); in TEST_F()
156 SoftmaxGrad(input_data, yt_data, out_data, sum_data, sum_mul, softmax_param); in TEST_F()
172 delete[] sum_data; in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dsoftmax.c39 static void SoftMaxP1Simple(const float *input_ptr, float *output_ptr, float *sum_data, int start, … in SoftMaxP1Simple() argument
54 sum_data[i] = _sum_data; in SoftMaxP1Simple()
58 void SoftMaxP1(const float *input_ptr, float *output_ptr, float *sum_data, int start, int count, in… in SoftMaxP1() argument
61 SoftMaxP1Simple(input_ptr, output_ptr, sum_data, start, count, length); in SoftMaxP1()
83 sum_data[k + sum_outter_offset] = _sum_data; in SoftMaxP1()
88 void SoftMaxP2(const float *input_ptr, float *output_ptr, const float *sum_data, int start, int cou… in SoftMaxP2() argument
97 output_ptr[inner_offset] = output_ptr[inner_offset] / sum_data[k + sum_outter_offset]; in SoftMaxP2()
Dsoftmax_grad.c21 void SoftmaxGrad(const float *input_ptr, const float *yt_ptr, float *output_ptr, float *sum_data, f… in SoftmaxGrad() argument
46 memset(sum_data, 0, (size_t)(inner_size) * sizeof(float)); in SoftmaxGrad()
51 sum_data[k] += output_ptr[offset] * input_ptr[offset]; in SoftmaxGrad()
57 *(output_ptr + outter_offset + k * N + j) += a * sum_data[j]; in SoftmaxGrad()
Dsoftmax.h24 void SoftMaxP1(const float *input_ptr, float *output_ptr, float *sum_data, int start, int count, in…
26 void SoftMaxP2(const float *input_ptr, float *output_ptr, const float *sum_data, int start, int cou…
Dsoftmax_grad.h41 void SoftmaxGrad(const float *input_ptr, const float *yt_ptr, float *output_ptr, float *sum_data, f…
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dsparse_softmax_cross_entropy_with_logits.cc93 float *sum_data = losses + data_size; in Execute() local
100 SoftMaxP1(ins, losses, sum_data, task_id * stride, count, length, inner_size_); in Execute()
103 SoftMaxP2(ins, losses, sum_data, task_id * stride, count, length, inner_size_); in Execute()
138 float *sum_data = losses + data_size; in Run() local
140 std::fill(sum_data, sum_data + sm_params_.input_shape_[0], 0.f); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dlog_softmax_fp32.c46 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter … in LogSoftmax() argument
65 sum_data[k + sum_outter_offset] = 0; in LogSoftmax()
73 sum_data[k + sum_outter_offset] += expf(output_ptr[axis_offset]); in LogSoftmax()
84 output_ptr[inner_offset] = output_ptr[inner_offset] - logf(sum_data[k + sum_outter_offset]); in LogSoftmax()
Dsoftmax_fp32.c105 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa… in Softmax() argument
124 sum_data[k + sum_outter_offset] = 0; in Softmax()
132 sum_data[k + sum_outter_offset] += output_ptr[axis_offset]; in Softmax()
143 output_ptr[inner_offset] = output_ptr[inner_offset] / sum_data[k + sum_outter_offset]; in Softmax()
Dlog_softmax_fp32.h25 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter …
Dsoftmax_fp32.h25 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dlog_softmax_fp16.c48 void LogSoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data, in LogSoftmaxFp16() argument
68 sum_data[k + sum_outter_offset] = 0; in LogSoftmaxFp16()
76 sum_data[k + sum_outter_offset] += exp(output_ptr[axis_offset]); in LogSoftmaxFp16()
87 output_ptr[inner_offset] = output_ptr[inner_offset] - log(sum_data[k + sum_outter_offset]); in LogSoftmaxFp16()
Dsoftmax_fp16.c94 void SoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data, in SoftmaxFp16() argument
114 sum_data[k + sum_outter_offset] = 0; in SoftmaxFp16()
122 sum_data[k + sum_outter_offset] += output_ptr[axis_offset]; in SoftmaxFp16()
133 output_ptr[inner_offset] = output_ptr[inner_offset] / sum_data[k + sum_outter_offset]; in SoftmaxFp16()
Dlog_softmax_fp16.h29 void LogSoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data,
Dsoftmax_fp16.h28 void SoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dsoftmax_int8.c20 …t SoftmaxInt8(const int8_t *input_ptr, int8_t *output_ptr, int count, int *exp_data, int *sum_data, in SoftmaxInt8() argument
55 sum_data[c] = exp_sum; in SoftmaxInt8()
61 int shifted_scale = ComputerReciprocal(sum_data[c], 12, &num_bits_over_unit); in SoftmaxInt8()
Dsoftmax_int8.h29 …t SoftmaxInt8(const int8_t *input_ptr, int8_t *output_ptr, int count, int *exp_data, int *sum_data,
/third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/operator_library/kernels/nnacl/fp32/
Dsoftmax_fp32.c105 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa… in Softmax() argument
131 sum_data[k + sum_outter_offset] += output_ptr[axis_offset]; in Softmax()
142 output_ptr[inner_offset] = output_ptr[inner_offset] / sum_data[k + sum_outter_offset]; in Softmax()
/third_party/vk-gl-cts/external/openglcts/modules/gl/
Dgl4cBufferStorageTests.cpp2999 GLuint sum_data[data_size]; in iterate() local
3014 sum_data[i] = sum; in iterate()
3076 if (0 != memcmp(destination_map.m_data, sum_data, data_size * sizeof(GLuint))) in iterate()
3174 GLuint sum_data[data_size]; in iterate() local
3189 sum_data[i] = sum; in iterate()
3251 if (0 != memcmp(destination_map.m_data, sum_data, data_size * sizeof(GLuint))) in iterate()