Searched defs:mean_data (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | batchnorm_tests.cc | 43 …float mean_data[] = {0.3016613, -0.89284, 0.63434774, 0.145766, 0.73353934, -0.6744012, 0.7087985,… in TEST_F() local
72 …float mean_data[] = {0.3016613, -0.89284, 0.63434774, 0.145766, 0.73353934, -0.6744012, 0.7087985}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | fused_batch_norm.c | 50 … const void *mean_data, const void *var_data, float eps, int kernel_num) { in FusedBatchNormCalculateScaleF32()
68 … const void *mean_data, const void *var_data, float eps, int kernel_num) { in FusedBatchNormCalculateScaleF16()
105 … const void *mean_data, const void *var_data, float eps, int kernel_num) { in FusedBatchNormBatchnorm2Scale()
|