| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | scale_fp16.c | 20 int outer_start, int outer_end, int axis_size, int inner_size) { in Fp16ScaleInner()
46 int outer_start, int outer_end, int axis_size) { in Fp16ScaleAxis()
71 int outer_start = task_id * outer_step; in DoScaleFp16() local
83 int outer_start, int outer_end, int axis_size, int inner_size) { in Fp16ScaleInnerRelu()
113 int outer_start, int outer_end, int axis_size) { in Fp16ScaleAxisRelu()
143 int outer_start = task_id * outer_step; in Fp16DoScaleRelu() local
155 int outer_start, int outer_end, int axis_size, int inner_size) { in Fp16ScaleInnerRelu6()
186 int outer_start, int outer_end, int axis_size) { in Fp16ScaleAxisRelu6()
217 int outer_start = task_id * outer_step; in DoScaleRelu6Fp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | scale_fp32.c | 20 …er(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, in ScaleInner()
55 …is(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, in ScaleAxis()
91 int outer_start = task_id * outer_step; in DoScale() local
102 …lu(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, in ScaleInnerRelu()
146 …lu(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, in ScaleAxisRelu()
191 int outer_start = task_id * outer_step; in DoScaleRelu() local
202 …u6(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, in ScaleInnerRelu6()
248 …u6(const float *in_data, float *out_data, const float *scale, const float *offset, int outer_start, in ScaleAxisRelu6()
295 int outer_start = task_id * outer_step; in DoScaleRelu6() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/ |
| D | cast_gather_reduce_fp32_sse.h | 33 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt64FusionSSE()
50 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt32FusionSSE()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/ |
| D | cast_gather_reduce_fp32_avx.h | 33 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt64FusionAVX()
50 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt32FusionAVX()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | cast_gather_reduce_fp32_neon.h | 32 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt64FusionNEON()
49 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt32FusionNEON()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/ |
| D | cast_gather_reduce_fp32_avx512.h | 33 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt64FusionAVX512()
50 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt32FusionAVX512()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/online_fusion/ |
| D | cast_gather_reduce_fp32.c | 22 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt64Fusion()
47 … int32_t inner_size, int32_t input_data_inner_size, int32_t outer_start, in Fp32CastGatherReduceInt32Fusion()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | stack_base.c | 18 void Stack(void **inputs, void *output, size_t input_num, size_t copy_size, int outer_start, int ou… in Stack()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/online_fusion/ |
| D | cast_gather_reduce_fp32.cc | 51 auto outer_start = task_id * outer_size_tile; in DoCastGatherReduceFusion() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | random_shuffle_cpu_kernel.cc | 136 auto outer_task = [this, output, input](size_t outer_start, size_t outer_end) { in TensorShuffleWithBatchRank()
|