Home
last modified time | relevance | path

Searched defs:scale_data (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dscale_tests.cc39 float scale_data[] = {1, 2}; in TEST_F() local
57 float scale_data[] = {1, 2, 3}; in TEST_F() local
75 float scale_data[] = {1, 2, -1}; in TEST_F() local
Dbatchnorm_tests.cc41 …float scale_data[] = {0.1201471, 0.142174, 0.5683258, 0.86815494, 0.23426804, 0.3634345, 0.0077846… in TEST_F() local
70 …float scale_data[] = {0.1201471, 0.142174, 0.5683258, 0.86815494, 0.23426804, 0.3634345, 0.0077846… in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dscale_fp32_tests.cc68 … float *input_data, float *scale_data, float *offset_data, float *output_data, int axis, in Prepare()
109 float scale_data[3] = {1.0, 2.0, 3.0}; in TEST_F() local
131 float scale_data[3] = {1.0, 2.0, 3.0}; in TEST_F() local
152 float scale_data[3] = {1.0, 2.0, 3.0}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dscale_int8.cc66int8_t *scale_data, const std::vector<int> &bias_shape, int8_t *bias_data, in Prepare()
106 int8_t scale_data[4] = {2, 2, 2, 2}; in TEST_F() local
132 int8_t scale_data[4] = {2, 2, 2, 2}; in TEST_F() local
158 int8_t scale_data[4] = {2, 2, 2, 2}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Dresize_coreml.cc49 auto scale_data = scale_tensor.Data().get(); in BuildLayer() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnapi/op/
Dinstance_norm_nnapi.cc34 auto scale_data = scale_tensor.MutableData(); in InitParams() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dfused_batch_norm.c49 void FusedBatchNormCalculateScaleF32(FusedBatchNormStruct *fbn, const void *scale_data, const void … in FusedBatchNormCalculateScaleF32()
67 void FusedBatchNormCalculateScaleF16(FusedBatchNormStruct *fbn, const void *scale_data, const void … in FusedBatchNormCalculateScaleF16()
104 int FusedBatchNormBatchnorm2Scale(FusedBatchNormStruct *fused_batch_norm, const void *scale_data, c… in FusedBatchNormBatchnorm2Scale()
Dgroup_norm.c90 const void *scale_data = groupnorm_stru->base.in_[C1NUM]->data_; in GroupNormImpl() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dscale.cc132 std::vector<char> scale_data(scale_shape.ElementsC4Num * FLT_size, 0); in InitWeights() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/
Dopencl_fusion.cc469 auto *scale_data = reinterpret_cast<float *>(scale->data()); in TryFusionConvScaleWeight() local