Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dreduce_fp32.c36 const float *outer_src = src_data + j * axis_size * inner_size; in ReduceMean() local
69 const int *outer_src = src_data + j * axis_size * inner_size; in IntReduceMean() local
114 const float *outer_src = src_data + j * axis_size * inner_size; in ReduceSum() local
155 const int *outer_src = src_data + j * axis_size * inner_size; in IntReduceSum() local
192 const float *outer_src = src_data + j * axis_size * inner_size; in ReduceMax() local
217 const int *outer_src = src_data + j * axis_size * inner_size; in IntReduceMax() local
242 const float *outer_src = src_data + j * axis_size * inner_size; in ReduceMin() local
267 const int *outer_src = src_data + j * axis_size * inner_size; in IntReduceMin() local
292 const bool *outer_src = src_data + j * axis_size * inner_size; in ReduceAll() local
317 const float *outer_src = src_data + j * axis_size * inner_size; in ReduceProd() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dreduce_int8.c183 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMeanInt8() local
217 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMeanLastAxis() local
258 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceSumInt8() local
290 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceSumLastAxis() local
330 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMaxLastAxis() local
366 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMaxInt8() local
390 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMinLastAxis() local
427 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMinInt8() local
449 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceProdLastAxis() local
492 const int32_t *outer_src = src_data + j * axis_size * inner_size; in ReduceProdInt8() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dreduce_fp16.c31 const float16_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMeanFp16() local
53 const float16_t *outer_src = src_data + j * axis_size * inner_size; in ReduceMaxFp16() local