Searched defs:sum_data (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | softmax.c | 39 static void SoftMaxP1Simple(const float *input_ptr, float *output_ptr, float *sum_data, int start, … in SoftMaxP1Simple()
58 void SoftMaxP1(const float *input_ptr, float *output_ptr, float *sum_data, int start, int count, in… in SoftMaxP1()
88 void SoftMaxP2(const float *input_ptr, float *output_ptr, const float *sum_data, int start, int cou… in SoftMaxP2()
|
| D | softmax_grad.c | 21 void SoftmaxGrad(const float *input_ptr, const float *yt_ptr, float *output_ptr, float *sum_data, f… in SoftmaxGrad()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | softmax_grad_fp32_tests.cc | 66 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
126 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
191 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
256 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
321 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sparse_softmax_cross_entropy_with_logits.cc | 93 float *sum_data = losses + data_size; in Execute() local
138 float *sum_data = losses + data_size; in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | softmax_int8.c | 20 …t SoftmaxInt8(const int8_t *input_ptr, int8_t *output_ptr, int count, int *exp_data, int *sum_data, in SoftmaxInt8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | log_softmax_fp32.c | 46 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter … in LogSoftmax()
|
| D | softmax_fp32.c | 105 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, const SoftmaxParameter *pa… in Softmax()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | log_softmax_fp16.c | 48 void LogSoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data, in LogSoftmaxFp16()
|
| D | softmax_fp16.c | 94 void SoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data, in SoftmaxFp16()
|
| /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()
|
| /third_party/vk-gl-cts/external/openglcts/modules/gl/ |
| D | gl4cBufferStorageTests.cpp | 2999 GLuint sum_data[data_size]; in iterate() local
3174 GLuint sum_data[data_size]; in iterate() local
|