| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | softmax_grad_utils.c | 40 static void SoftMaxP1Simple(const float *input_ptr, float *output_ptr, float *sum_data, int start, … in SoftMaxP1Simple()
59 void SoftMaxP1(const float *input_ptr, float *output_ptr, float *sum_data, int start, int count, in… in SoftMaxP1()
89 void SoftMaxP2(const float *input_ptr, float *output_ptr, const float *sum_data, int start, int cou… in SoftMaxP2()
|
| D | softmax_grad.c | 20 void SoftmaxGrad(const float *input_ptr, const float *yt_ptr, float *output_ptr, float *sum_data, f… in SoftmaxGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | softmax_grad_fp32_tests.cc | 48 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
113 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
183 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
253 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
322 float *sum_data = new (std::nothrow) float[inner_size]; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | sparse_softmax_cross_entropy_with_logits.cc | 93 float *sum_data = losses + data_size; in DoExecute() local
140 float *sum_data = losses + data_size; in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | softmax_cpu_kernel.cc | 44 void SoftmaxFunc(const T *input_ptr, T *output_ptr, acc_T *sum_data, int start, int end, int dim_ax… in SoftmaxFunc()
155 auto sum_data = GetDeviceAddress<acc_T>(workspace, kIndex0); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | softmax_int8.c | 20 …xInt8(const int8_t *input_ptr, int8_t *output_ptr, int count, int32_t *exp_data, int32_t *sum_data, in SoftmaxInt8()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | log_softmax_fp32.c | 46 void LogSoftmax(const float *input_ptr, float *output_ptr, float *sum_data, int32_t *input_shape, i… in LogSoftmax()
|
| D | softmax_fp32.c | 85 void Softmax(const float *input_ptr, float *output_ptr, float *sum_data, int axis, int n_dim, in Softmax()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | log_softmax_fp16.c | 48 void LogSoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data, int *in… in LogSoftmaxFp16()
|
| D | softmax_fp16.c | 94 void SoftmaxFp16(const float16_t *input_ptr, float16_t *output_ptr, float16_t *sum_data, int axis, … in SoftmaxFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/ |
| D | meta_graph_serializer.cc | 112 auto sum_data = reinterpret_cast<uint32_t *>(head_data); in Init() local
|
| /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
|