Home
last modified time | relevance | path

Searched defs:outer_step (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dscale_fp16.c70 int outer_step = UP_DIV(scale_param->outer_size_, scale_param->base_.thread_nr_); in DoScaleFp16() local
142 int outer_step = UP_DIV(scale_param->outer_size_, scale_param->base_.thread_nr_); in Fp16DoScaleRelu() local
216 int outer_step = UP_DIV(scale_param->outer_size_, scale_param->base_.thread_nr_); in DoScaleRelu6Fp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dscale_fp32.c90 int outer_step = UP_DIV(scale_param->outer_size_, scale_param->base_.thread_nr_); in DoScale() local
190 int outer_step = UP_DIV(scale_param->outer_size_, scale_param->base_.thread_nr_); in DoScaleRelu() local
294 int outer_step = UP_DIV(scale_param->outer_size_, scale_param->base_.thread_nr_); in DoScaleRelu6() local