| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | reduce_fp32_simd.h.in | 31 const float *inner_src = outer_src + index; variable
53 const float *inner_src = outer_src + index; variable
66 const float *inner_src = outer_src + index; variable
79 const float *inner_src = outer_src + index; variable
101 const float *inner_src = outer_src + index; variable
114 const float *inner_src = outer_src + index; variable
127 const float *inner_src = outer_src + index; variable
140 const int32_t *inner_src = outer_src + index; variable
153 const int32_t *inner_src = outer_src + index; variable
166 const int32_t *inner_src = outer_src + index; variable
[all …]
|
| D | reduce_fp32.c | 175 const bool *inner_src = outer_src + k; in ReduceAll() local
200 const int32_t *inner_src = outer_src + k; in IntReduceProd() local
294 const float *inner_src = src_data + k; in ReduceSumDim2Axis0() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | reduce_fp32_neon.h | 34 const float *inner_src = outer_src + index; in ReduceSumNEON() local
56 const float *inner_src = outer_src + index; in ReduceMeanNEON() local
69 const float *inner_src = outer_src + index; in ReduceMinNEON() local
82 const float *inner_src = outer_src + index; in ReduceMaxNEON() local
104 const float *inner_src = outer_src + index; in ReduceProdNEON() local
117 const float *inner_src = outer_src + index; in ReduceSumSquareNEON() local
130 const float *inner_src = outer_src + index; in ReduceL2NormNEON() local
143 const int32_t *inner_src = outer_src + index; in IntReduceSumNEON() local
156 const int32_t *inner_src = outer_src + index; in IntReduceMeanNEON() local
169 const int32_t *inner_src = outer_src + index; in IntReduceMinNEON() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/ |
| D | reduce_fp32_avx512.h | 35 const float *inner_src = outer_src + index; in ReduceSumAVX512() local
57 const float *inner_src = outer_src + index; in ReduceMeanAVX512() local
70 const float *inner_src = outer_src + index; in ReduceMinAVX512() local
83 const float *inner_src = outer_src + index; in ReduceMaxAVX512() local
105 const float *inner_src = outer_src + index; in ReduceProdAVX512() local
118 const float *inner_src = outer_src + index; in ReduceSumSquareAVX512() local
131 const float *inner_src = outer_src + index; in ReduceL2NormAVX512() local
144 const int32_t *inner_src = outer_src + index; in IntReduceSumAVX512() local
157 const int32_t *inner_src = outer_src + index; in IntReduceMeanAVX512() local
170 const int32_t *inner_src = outer_src + index; in IntReduceMinAVX512() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/ |
| D | reduce_fp32_avx.h | 35 const float *inner_src = outer_src + index; in ReduceSumAVX() local
57 const float *inner_src = outer_src + index; in ReduceMeanAVX() local
70 const float *inner_src = outer_src + index; in ReduceMinAVX() local
83 const float *inner_src = outer_src + index; in ReduceMaxAVX() local
105 const float *inner_src = outer_src + index; in ReduceProdAVX() local
118 const float *inner_src = outer_src + index; in ReduceSumSquareAVX() local
131 const float *inner_src = outer_src + index; in ReduceL2NormAVX() local
144 const int32_t *inner_src = outer_src + index; in IntReduceSumAVX() local
157 const int32_t *inner_src = outer_src + index; in IntReduceMeanAVX() local
170 const int32_t *inner_src = outer_src + index; in IntReduceMinAVX() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/ |
| D | reduce_fp32_sse.h | 35 const float *inner_src = outer_src + index; in ReduceSumSSE() local
57 const float *inner_src = outer_src + index; in ReduceMeanSSE() local
70 const float *inner_src = outer_src + index; in ReduceMinSSE() local
83 const float *inner_src = outer_src + index; in ReduceMaxSSE() local
105 const float *inner_src = outer_src + index; in ReduceProdSSE() local
118 const float *inner_src = outer_src + index; in ReduceSumSquareSSE() local
131 const float *inner_src = outer_src + index; in ReduceL2NormSSE() local
144 const int32_t *inner_src = outer_src + index; in IntReduceSumSSE() local
157 const int32_t *inner_src = outer_src + index; in IntReduceMeanSSE() local
170 const int32_t *inner_src = outer_src + index; in IntReduceMinSSE() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | reduce_fp16.c | 36 const float16_t *inner_src = outer_src + k; in ReduceMeanFp16() local
58 const float16_t *inner_src = outer_src + k; in ReduceMaxFp16() local
83 const float16_t *inner_src = outer_src + k; in ReduceMinFp16() local
108 const float16_t *inner_src = outer_src + k; in ReduceProdFp16() local
138 const float16_t *inner_src = src_data + j; in ReduceSumFp16() local
148 const float16_t *inner_src = src_data + j; in ReduceSumFp16() local
178 const float16_t *inner_src = src_data + j; in ReduceL2NormFp16() local
189 const float16_t *inner_src = src_data + j; in ReduceL2NormFp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | bias_add_grad_cpu_kernel.cc | 96 const T *inner_src = input_addr + offset; in LaunchKernel() local
107 const T *inner_src = tmp_res.data() + start + k; in LaunchKernel() local
120 const T *inner_src = input_addr + start + k; in LaunchKernel() local
|
| D | bias_add_cpu_kernel.cc | 143 const T *inner_src = src_addr + n_offset; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | reduce_int8.c | 186 const int32_t *inner_src = outer_src + k; in ReduceMeanInt8() local
220 const int32_t *inner_src = outer_src + k; in ReduceMeanLastAxis() local
261 const int32_t *inner_src = outer_src + k; in ReduceSumInt8() local
293 const int32_t *inner_src = outer_src + k; in ReduceSumLastAxis() local
333 const int32_t *inner_src = outer_src + k; in ReduceMaxLastAxis() local
369 const int32_t *inner_src = outer_src + k; in ReduceMaxInt8() local
393 const int32_t *inner_src = outer_src + k; in ReduceMinLastAxis() local
430 const int32_t *inner_src = outer_src + k; in ReduceMinInt8() local
452 const int32_t *inner_src = outer_src + k; in ReduceProdLastAxis() local
495 const int32_t *inner_src = outer_src + k; in ReduceProdInt8() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | bias_add_grad.cc | 127 const T *inner_src = input_x + start + k; in BiasAddGradCompute() local
|