| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | reduce_fp32_neon.h | 31 static inline int64_t ReduceSumNEON(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceSumNEON()
53 static inline int64_t ReduceMeanNEON(int64_t index, const float *outer_src, float *outer_dst, int i… in ReduceMeanNEON()
66 static inline int64_t ReduceMinNEON(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceMinNEON()
79 static inline int64_t ReduceMaxNEON(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceMaxNEON()
101 static inline int64_t ReduceProdNEON(int64_t index, const float *outer_src, float *outer_dst, int i… in ReduceProdNEON()
114 static inline int64_t ReduceSumSquareNEON(int64_t index, const float *outer_src, float *outer_dst, … in ReduceSumSquareNEON()
127 static inline int64_t ReduceL2NormNEON(int64_t index, const float *outer_src, float *outer_dst, int… in ReduceL2NormNEON()
140 static inline int64_t IntReduceSumNEON(int64_t index, const int32_t *outer_src, int32_t *outer_dst,… in IntReduceSumNEON()
153 static inline int64_t IntReduceMeanNEON(int64_t index, const int32_t *outer_src, int32_t *outer_dst… in IntReduceMeanNEON()
166 static inline int64_t IntReduceMinNEON(int64_t index, const int32_t *outer_src, int32_t *outer_dst,… in IntReduceMinNEON()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/ |
| D | reduce_fp32_avx512.h | 32 static inline int64_t ReduceSumAVX512(int64_t index, const float *outer_src, float *outer_dst, int … in ReduceSumAVX512()
54 static inline int64_t ReduceMeanAVX512(int64_t index, const float *outer_src, float *outer_dst, int… in ReduceMeanAVX512()
67 static inline int64_t ReduceMinAVX512(int64_t index, const float *outer_src, float *outer_dst, int … in ReduceMinAVX512()
80 static inline int64_t ReduceMaxAVX512(int64_t index, const float *outer_src, float *outer_dst, int … in ReduceMaxAVX512()
102 static inline int64_t ReduceProdAVX512(int64_t index, const float *outer_src, float *outer_dst, int… in ReduceProdAVX512()
115 static inline int64_t ReduceSumSquareAVX512(int64_t index, const float *outer_src, float *outer_dst… in ReduceSumSquareAVX512()
128 static inline int64_t ReduceL2NormAVX512(int64_t index, const float *outer_src, float *outer_dst, i… in ReduceL2NormAVX512()
141 …nt64_t IntReduceSumAVX512(int64_t index, const int32_t *outer_src, int32_t *outer_dst, int inner_s… in IntReduceSumAVX512()
154 …t64_t IntReduceMeanAVX512(int64_t index, const int32_t *outer_src, int32_t *outer_dst, int inner_s… in IntReduceMeanAVX512()
167 …nt64_t IntReduceMinAVX512(int64_t index, const int32_t *outer_src, int32_t *outer_dst, int inner_s… in IntReduceMinAVX512()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/ |
| D | reduce_fp32_avx.h | 32 static inline int64_t ReduceSumAVX(int64_t index, const float *outer_src, float *outer_dst, int inn… in ReduceSumAVX()
54 static inline int64_t ReduceMeanAVX(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceMeanAVX()
67 static inline int64_t ReduceMinAVX(int64_t index, const float *outer_src, float *outer_dst, int inn… in ReduceMinAVX()
80 static inline int64_t ReduceMaxAVX(int64_t index, const float *outer_src, float *outer_dst, int inn… in ReduceMaxAVX()
102 static inline int64_t ReduceProdAVX(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceProdAVX()
115 static inline int64_t ReduceSumSquareAVX(int64_t index, const float *outer_src, float *outer_dst, i… in ReduceSumSquareAVX()
128 static inline int64_t ReduceL2NormAVX(int64_t index, const float *outer_src, float *outer_dst, int … in ReduceL2NormAVX()
141 static inline int64_t IntReduceSumAVX(int64_t index, const int32_t *outer_src, int32_t *outer_dst, … in IntReduceSumAVX()
154 static inline int64_t IntReduceMeanAVX(int64_t index, const int32_t *outer_src, int32_t *outer_dst,… in IntReduceMeanAVX()
167 static inline int64_t IntReduceMinAVX(int64_t index, const int32_t *outer_src, int32_t *outer_dst, … in IntReduceMinAVX()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/ |
| D | reduce_fp32_sse.h | 32 static inline int64_t ReduceSumSSE(int64_t index, const float *outer_src, float *outer_dst, int inn… in ReduceSumSSE()
54 static inline int64_t ReduceMeanSSE(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceMeanSSE()
67 static inline int64_t ReduceMinSSE(int64_t index, const float *outer_src, float *outer_dst, int inn… in ReduceMinSSE()
80 static inline int64_t ReduceMaxSSE(int64_t index, const float *outer_src, float *outer_dst, int inn… in ReduceMaxSSE()
102 static inline int64_t ReduceProdSSE(int64_t index, const float *outer_src, float *outer_dst, int in… in ReduceProdSSE()
115 static inline int64_t ReduceSumSquareSSE(int64_t index, const float *outer_src, float *outer_dst, i… in ReduceSumSquareSSE()
128 static inline int64_t ReduceL2NormSSE(int64_t index, const float *outer_src, float *outer_dst, int … in ReduceL2NormSSE()
141 static inline int64_t IntReduceSumSSE(int64_t index, const int32_t *outer_src, int32_t *outer_dst, … in IntReduceSumSSE()
154 static inline int64_t IntReduceMeanSSE(int64_t index, const int32_t *outer_src, int32_t *outer_dst,… in IntReduceMeanSSE()
167 static inline int64_t IntReduceMinSSE(int64_t index, const int32_t *outer_src, int32_t *outer_dst, … in IntReduceMinSSE()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/ |
| D | reduce_int8.c | 184 int32_t *outer_dst = dst_data + j * inner_size; in ReduceMeanInt8() local
218 int8_t *outer_dst = dst_data + j * inner_size; in ReduceMeanLastAxis() local
259 int32_t *outer_dst = dst_data + j * inner_size; in ReduceSumInt8() local
291 int8_t *outer_dst = dst_data + j * inner_size; in ReduceSumLastAxis() local
331 int8_t *outer_dst = dst_data + j * inner_size; in ReduceMaxLastAxis() local
367 int32_t *outer_dst = dst_data + j * inner_size; in ReduceMaxInt8() local
391 int8_t *outer_dst = dst_data + j * inner_size; in ReduceMinLastAxis() local
428 int32_t *outer_dst = dst_data + j * inner_size; in ReduceMinInt8() local
450 int8_t *outer_dst = dst_data + j * inner_size; in ReduceProdLastAxis() local
493 int32_t *outer_dst = dst_data + j * inner_size; in ReduceProdInt8() local
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | reduce_fp16.c | 34 float16_t *outer_dst = dst_data + j * inner_size; in ReduceMeanFp16() local
56 float16_t *outer_dst = dst_data + j * inner_size; in ReduceMaxFp16() local
81 float16_t *outer_dst = dst_data + j * inner_size; in ReduceMinFp16() local
106 float16_t *outer_dst = dst_data + j * inner_size; in ReduceProdFp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | reduce_fp32.c | 27 #define ReduceCoreCalc(op_name, op_type, outer_src, outer_dst, k) \ argument
173 bool *outer_dst = dst_data + j * inner_size; in ReduceAll() local
198 int32_t *outer_dst = dst_data + j * inner_size; in IntReduceProd() local
|