Home
last modified time | relevance | path

Searched defs:inner_src (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c39 const float *inner_src = outer_src + k; in ReduceMean() local
74 const int *inner_src = outer_src + k; in IntReduceMean() local
88 const int *inner_src = outer_src + k; in IntReduceMean() local
119 const float *inner_src = outer_src + k; in ReduceSum() local
129 const float *inner_src = outer_src + k; in ReduceSum() local
160 const int *inner_src = outer_src + k; in IntReduceSum() local
170 const int *inner_src = outer_src + k; in IntReduceSum() local
195 const float *inner_src = outer_src + k; in ReduceMax() local
220 const int *inner_src = outer_src + k; in IntReduceMax() local
245 const float *inner_src = outer_src + k; in ReduceMin() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dreduce_fp16.c34 const float16_t *inner_src = outer_src + k; in ReduceMeanFp16() local
56 const float16_t *inner_src = outer_src + k; in ReduceMaxFp16() local
86 const float16_t *inner_src = src_data + j; in ReduceSumFp16() local
96 const float16_t *inner_src = src_data + j; in ReduceSumFp16() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dreduce_int8.c186 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 …]