Home
last modified time | relevance | path

Searched defs:outer_end (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dscale_fp16.c20 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()
72 int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); 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()
144 int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); 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()
218 int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); in DoScaleRelu6Fp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dscale_fp32.c21 int outer_end, int axis_size, int inner_size) { in ScaleInner()
56 int outer_end, int axis_size) { in ScaleAxis()
92 int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); in DoScale() local
103 int outer_end, int axis_size, int inner_size) { in ScaleInnerRelu()
147 int outer_end, int axis_size) { in ScaleAxisRelu()
192 int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); in DoScaleRelu() local
203 int outer_end, int axis_size, int inner_size) { in ScaleInnerRelu6()
249 int outer_end, int axis_size) { in ScaleAxisRelu6()
296 int outer_end = MSMIN(outer_start + outer_step, scale_param->outer_size_); in DoScaleRelu6() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/
Dcast_gather_reduce_fp32_sse.h34 int32_t outer_end) { in Fp32CastGatherReduceInt64FusionSSE()
51 int32_t outer_end) { in Fp32CastGatherReduceInt32FusionSSE()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/
Dcast_gather_reduce_fp32_avx.h34 int32_t outer_end) { in Fp32CastGatherReduceInt64FusionAVX()
51 int32_t outer_end) { in Fp32CastGatherReduceInt32FusionAVX()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/
Dcast_gather_reduce_fp32_neon.h33 int32_t outer_end) { in Fp32CastGatherReduceInt64FusionNEON()
50 int32_t outer_end) { in Fp32CastGatherReduceInt32FusionNEON()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/
Dcast_gather_reduce_fp32_avx512.h34 int32_t outer_end) { in Fp32CastGatherReduceInt64FusionAVX512()
51 int32_t outer_end) { in Fp32CastGatherReduceInt32FusionAVX512()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/online_fusion/
Dcast_gather_reduce_fp32.c23 int32_t outer_end) { in Fp32CastGatherReduceInt64Fusion()
48 int32_t outer_end) { in Fp32CastGatherReduceInt32Fusion()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dstack_base.c18 …(void **inputs, void *output, size_t input_num, size_t copy_size, int outer_start, int outer_end) { in Stack()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/online_fusion/
Dcast_gather_reduce_fp32.cc52 auto outer_end = MSMIN(outer_size_, outer_size_tile + outer_start); in DoCastGatherReduceFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Drandom_shuffle_cpu_kernel.cc136 auto outer_task = [this, output, input](size_t outer_start, size_t outer_end) { in TensorShuffleWithBatchRank()
/third_party/wpa_supplicant/wpa_supplicant-2.9/src/eap_peer/
Deap_teap.c1539 const u8 *outer_pos, *outer_end; in eap_teap_process_start() local
/third_party/wpa_supplicant/wpa_supplicant-2.9_standard/src/eap_peer/
Deap_teap.c1658 const u8 *outer_pos, *outer_end; in eap_teap_process_start() local