Home
last modified time | relevance | path

Searched refs:bias_data (Results 1 – 25 of 56) sorted by relevance

123

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dconv2d_tests.cc39 …d TestMain_Conv2D(const std::string &attr, float *input_data, float *weight_data, float *bias_data, in TestMain_Conv2D() argument
48 if (bias_data) { in TestMain_Conv2D()
49 input_infos.emplace_back(bias_shape, bias_data, CONST_TENSOR); in TestMain_Conv2D()
60 float bias_data[] = {0, 0}; in TEST_F() local
63 TestMain_Conv2D(attr, input_data, weight_data, bias_data, output_data, ActType_No, false, 1e-3f); in TEST_F()
64 TestMain_Conv2D(attr, input_data, weight_data, bias_data, output_data, ActType_No, true, 1e-6f); in TEST_F()
67 …TestMain_Conv2D(attr, input_data, weight_data, bias_data, output_data_relu, ActType_Relu, false, 1… in TEST_F()
68 …TestMain_Conv2D(attr, input_data, weight_data, bias_data, output_data_relu, ActType_Relu, true, 1e… in TEST_F()
71 …TestMain_Conv2D(attr, input_data, weight_data, bias_data, output_data_relu6, ActType_Relu6, false,… in TEST_F()
72 …TestMain_Conv2D(attr, input_data, weight_data, bias_data, output_data_relu6, ActType_Relu6, true, … in TEST_F()
[all …]
Dconv2d_transpose_tests.cc64 float bias_data[] = {0.5}; in TEST_F() local
73 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
91 float bias_data[] = {0.5}; in TEST_F() local
101 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
119 float bias_data[] = {0.5}; in TEST_F() local
129 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
147 float bias_data[] = {0.5}; in TEST_F() local
157 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
176 float bias_data[] = {0.5}; in TEST_F() local
188 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
[all …]
Dfullconnection_tests.cc60 float bias_data[] = {1, 1, 1}; variable
68 {bias_shape, bias_data, CONST_TENSOR}},
82 float bias_data[] = {1, 1}; variable
90 {bias_shape, bias_data, CONST_TENSOR}},
105 float bias_data[] = {1, 1, 1, 1}; variable
113 {bias_shape, bias_data, CONST_TENSOR}},
128 float bias_data[] = {1, 1, 1, 1}; variable
134 …_shape, input_data, VAR}, {weight_shape, weight_data, VAR}, {bias_shape, bias_data, CONST_TENSOR}},
Ddepthwise_conv2d_tests.cc74 float bias_data[] = {0, 0, 0, 0}; in TEST_F() local
88 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
122 float bias_data[] = {0, 0, 0, 0, 0}; in TEST_F() local
135 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
165 float bias_data[] = {0, 0, 0, 0}; in TEST_F() local
180 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
226 float bias_data[] = {0, 0, 0, 0, 0, 0}; in TEST_F() local
249 {bias_shape, bias_data, CONST_TENSOR}}, in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dwinograd_utils.h31 typedef void (*OutputTransFunc)(const float *src_data, float *dst_data, const float *bias_data, int…
228 void OutputTransform4x2Unit(const float *src_data, float *dst_data, const float *bias_data, int src…
230 void OutputTransform4x2ReluUnit(const float *src_data, float *dst_data, const float *bias_data, int…
232 void OutputTransform4x2Relu6Unit(const float *src_data, float *dst_data, const float *bias_data, in…
234 void OutputTransform4x3Unit(const float *src_data, float *dst_data, const float *bias_data, int src…
236 void OutputTransform4x3ReluUnit(const float *src_data, float *dst_data, const float *bias_data, int…
238 void OutputTransform4x3Relu6Unit(const float *src_data, float *dst_data, const float *bias_data, in…
241 void OutputTransform6x2Unit(const float *src_data, float *dst_data, const float *bias_data, int src…
243 void OutputTransform6x2ReluUnit(const float *src_data, float *dst_data, const float *bias_data, int…
245 void OutputTransform6x2Relu6Unit(const float *src_data, float *dst_data, const float *bias_data, in…
[all …]
Dwinograd_avx.h26 typedef void (*OutputTransFunc)(const float *src_data, float *dst_data, const float *bias_data, int…
213 void OutputTransform4x2AvxUnit(const float *src_data, float *dst_data, const float *bias_data, int …
215 void OutputTransform4x2ReluAvxUnit(const float *src_data, float *dst_data, const float *bias_data, …
217 void OutputTransform4x2Relu6AvxUnit(const float *src_data, float *dst_data, const float *bias_data,…
219 void OutputTransform4x3AvxUnit(const float *src_data, float *dst_data, const float *bias_data, int …
221 void OutputTransform4x3ReluAvxUnit(const float *src_data, float *dst_data, const float *bias_data, …
223 void OutputTransform4x3Relu6AvxUnit(const float *src_data, float *dst_data, const float *bias_data,…
226 void OutputTransform6x2AvxUnit(const float *src_data, float *dst_data, const float *bias_data, int …
228 void OutputTransform6x2ReluAvxUnit(const float *src_data, float *dst_data, const float *bias_data, …
230 void OutputTransform6x2Relu6AvxUnit(const float *src_data, float *dst_data, const float *bias_data,…
[all …]
Dconv_depthwise_fp32.h33 …nvDw(float *output_data, const float *input_data, const float *weight_data, const float *bias_data,
48 …Fp32(float *output_data, const float *input_data, const float *weight_data, const float *bias_data,
103 …Fp32(float *output_data, const float *input_data, const float *weight_data, const float *bias_data,
123 void ConvDw3x3Line(float *dst, float **lines, const float *weight, const float *bias_data, int widt…
126 const float *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh);
132 …tion(float *output_data, float **indirect_buffer, const float *weight_data, const float *bias_data,
135 …Fp32(float *output_data, const float *input_data, const float *weight_data, const float *bias_data,
Dwinograd_avx.c20 void OutputTransform4x2AvxUnit(const float *src_data, float *dst_data, const float *bias_data, int … in OutputTransform4x2AvxUnit() argument
26 MS_FLOAT32X8 bias_ptr = MS_LD256_F32(bias_data); in OutputTransform4x2AvxUnit()
52 void OutputTransform4x2ReluAvxUnit(const float *src_data, float *dst_data, const float *bias_data, … in OutputTransform4x2ReluAvxUnit() argument
59 MS_FLOAT32X8 bias_ptr = MS_LD256_F32(bias_data); in OutputTransform4x2ReluAvxUnit()
87 void OutputTransform4x2Relu6AvxUnit(const float *src_data, float *dst_data, const float *bias_data,… in OutputTransform4x2Relu6AvxUnit() argument
95 MS_FLOAT32X8 bias_ptr = MS_LD256_F32(bias_data); in OutputTransform4x2Relu6AvxUnit()
125 void OutputTransform4x3AvxUnit(const float *src_data, float *dst_data, const float *bias_data, int … in OutputTransform4x3AvxUnit() argument
131 MS_FLOAT32X8 bias_ptr = MS_LD256_F32(bias_data); in OutputTransform4x3AvxUnit()
161 void OutputTransform4x3ReluAvxUnit(const float *src_data, float *dst_data, const float *bias_data, … in OutputTransform4x3ReluAvxUnit() argument
168 MS_FLOAT32X8 bias_ptr = MS_LD256_F32(bias_data); in OutputTransform4x3ReluAvxUnit()
[all …]
Dconv_winograd_fp32.c24 void ConvWinogardFp32(const float *input_data, const float *trans_weight, const float *bias_data, f… in ConvWinogardFp32() argument
90 …WinogradOutputNHWCTransform(dst_ptr, output_ptr, bias_data, cal_num, out_tile_index, out_w_block, … in ConvWinogardFp32()
94 …WinogradOutputNC4HW4Transform(dst_ptr, output_ptr, bias_data, cal_num, out_tile_index, out_w_block… in ConvWinogardFp32()
97 …WinogradOutputNHWCTransform(dst_ptr, output_ptr, bias_data, cal_num, out_tile_index, out_w_block, … in ConvWinogardFp32()
Dwinograd_utils.c327 void OutputTransform4x2Unit(const float *src_data, float *dst_data, const float *bias_data, int src… in OutputTransform4x2Unit() argument
334 MS_FLOAT32X4 bias_ptr = MS_LDQ_F32(bias_data); in OutputTransform4x2Unit()
382 dst_data[i + dst_k_offset + j * out_c] = m[j + m_k_offset] + bias_data[i]; in OutputTransform4x2Unit()
389 void OutputTransform4x2ReluUnit(const float *src_data, float *dst_data, const float *bias_data, int… in OutputTransform4x2ReluUnit() argument
397 MS_FLOAT32X4 bias_ptr = MS_LDQ_F32(bias_data); in OutputTransform4x2ReluUnit()
447 float out_value = m[j + m_k_offset] + bias_data[i]; in OutputTransform4x2ReluUnit()
456 void OutputTransform4x2Relu6Unit(const float *src_data, float *dst_data, const float *bias_data, in… in OutputTransform4x2Relu6Unit() argument
465 MS_FLOAT32X4 bias_ptr = MS_LDQ_F32(bias_data); in OutputTransform4x2Relu6Unit()
517 float out_value = m[j + m_k_offset] + bias_data[i]; in OutputTransform4x2Relu6Unit()
527 void OutputTransform4x3Unit(const float *src_data, float *dst_data, const float *bias_data, int src… in OutputTransform4x3Unit() argument
[all …]
Dwinograd_transform.c98 void WinogradOutputNHWCTransform(const float *gemm_out, float *out_data, const float *bias_data, in… in WinogradOutputNHWCTransform() argument
136 const float *bias_ptr = bias_data + j * C4NUM; in WinogradOutputNHWCTransform()
148 const float *bias_ptr = bias_data + j * C8NUM; in WinogradOutputNHWCTransform()
157 void WinogradOutputNC4HW4Transform(const float *gemm_out, float *out_data, const float *bias_data, … in WinogradOutputNC4HW4Transform() argument
193 const float *bias_ptr = bias_data + j * C8NUM; in WinogradOutputNC4HW4Transform()
206 const float *bias_ptr = bias_data + j * C4NUM; in WinogradOutputNC4HW4Transform()
Dwinograd_transform.h35 void WinogradOutputNHWCTransform(const float *gemm_out, float *out_data, const float *bias_data, in…
39 void WinogradOutputNC4HW4Transform(const float *gemm_out, float *out_data, const float *bias_data, …
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_utils_fp16.h32 … (*OutputTransFp16Func)(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
413 …putTransform4x2UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
415 …ransform4x2ReluUnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
417 …ansform4x2Relu6UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
419 …putTransform4x3UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
421 …ransform4x3ReluUnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
423 …ansform4x3Relu6UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
426 …putTransform6x2UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
428 …ransform6x2ReluUnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
430 …ansform6x2Relu6UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data,
[all …]
Dconv_depthwise_fp16.h44 const float16_t *bias_data, const ConvParameter *conv_param, int task_id);
47 … const float16_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
51 … const float16_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
55 …3x3LineFp16(float16_t *dst, float16_t **lines, const float16_t *weight, const float16_t *bias_data,
58 … const float16_t *bias_data, const ConvParameter *conv_param, int start_oh, int end_oh);
Dconv_fp16.h34 … const float16_t *bias_data, float16_t *col_major_input, float16_t *output_data, int task_id,
38 … const float16_t *bias_data, float16_t *col_major_input, float16_t *output_data, int task_id,
50 …inogardFp16(const float16_t *input_data, const float16_t *trans_weight, const float16_t *bias_data,
Dconv_fp16.c24 … const float16_t *bias_data, float16_t *col_major_input, float16_t *output_data, int task_id, in ConvFp16() argument
59 …MatMulFp16(col_major_input, packed_weight, output_data + out_offset, bias_data, conv_param->act_ty… in ConvFp16()
66 … const float16_t *bias_data, float16_t *col_major_input, float16_t *output_data, int task_id, in ConvOutNc8hw8Fp16() argument
101 const float16_t *cur_bias = bias_data; in ConvOutNc8hw8Fp16()
188 …inogardFp16(const float16_t *input_data, const float16_t *trans_weight, const float16_t *bias_data, in ConvWinogardFp16() argument
248 …tNHWCTransformFp16(gemm_out + task_id * gemm_out_offset, output_data + out_batch_offset, bias_data, in ConvWinogardFp16()
252 … bias_data, cal_num, out_tile_index, out_w_block, conv_param, out_func); in ConvWinogardFp16()
Dwinograd_transform_fp16.h36 …utNHWCTransformFp16(const float16_t *gemm_out, float16_t *tmp_out_data, const float16_t *bias_data,
40 …NC8HW8TransformFp16(const float16_t *gemm_out, float16_t *tmp_out_data, const float16_t *bias_data,
Dwinograd_utils_fp16.c462 …putTransform4x2UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data, in OutputTransform4x2UnitFp16() argument
470 float16x8_t bias_ptr = vld1q_f16(bias_data); in OutputTransform4x2UnitFp16()
498 float16x4_t bias_ptr = vld1_f16(bias_data); in OutputTransform4x2UnitFp16()
531 float16_t bias_ptr = bias_data[z]; in OutputTransform4x2UnitFp16()
552 …ransform4x2ReluUnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data, in OutputTransform4x2ReluUnitFp16() argument
561 float16x8_t bias_ptr = vld1q_f16(bias_data); in OutputTransform4x2ReluUnitFp16()
592 float16x4_t bias_ptr = vld1_f16(bias_data); in OutputTransform4x2ReluUnitFp16()
627 float16_t bias_ptr = bias_data[z]; in OutputTransform4x2ReluUnitFp16()
650 …ansform4x2Relu6UnitFp16(const float16_t *src_data, float16_t *dst_data, const float16_t *bias_data, in OutputTransform4x2Relu6UnitFp16() argument
660 float16x8_t bias_ptr = vld1q_f16(bias_data); in OutputTransform4x2Relu6UnitFp16()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.h28 const int32_t *bias_data, const ConvParameter *conv_param, int task_id);
31 … const int32_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding);
34 … const int32_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
37 …nt8_t *output_data, const int8_t *input_data, const int16_t *weight_data, const int32_t *bias_data,
42 … const int32_t *bias_data, const ConvParameter *conv_param, const SlidingWindowParam *sliding,
Dconv_int8.c20 …const int32_t *bias_data, int8_t *output_data, int32_t *filter_zp, int32_t *input_sum, int task_id, in ConvInt8() argument
70 …_input, packed_weight, gemm_output, real_cal_num, out_channel, unit_size, tmp_input_sum, bias_data, in ConvInt8()
77 tmp_input_sum, bias_data, conv_param->conv_quant_arg_.left_shift_, in ConvInt8()
83 UP_ROUND(out_channel, C4NUM), unit_size, tmp_input_sum, bias_data, in ConvInt8()
92 … bias_data, conv_param->conv_quant_arg_.left_shift_, conv_param->conv_quant_arg_.right_shift_, in ConvInt8()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_transform_fusion.cc218 float *bias_data = nullptr; in GenNewConvTensor() local
224 bias_data = reinterpret_cast<float *>(bias_tensor->data_c()); in GenNewConvTensor()
227 bias_data = new (std::nothrow) float[kernel_num]; in GenNewConvTensor()
228 if (bias_data == nullptr) { in GenNewConvTensor()
232 if (memset_s(bias_data, kernel_num * sizeof(float), 0, kernel_num * sizeof(float)) != EOK) { in GenNewConvTensor()
233 delete[] bias_data; in GenNewConvTensor()
237 if (CalNewBiasTensor(bias_data, kernel_num, bias_flag, trans_scale, trans_bias) != lite::RET_OK) { in GenNewConvTensor()
240 delete[] bias_data; in GenNewConvTensor()
245 auto bias_node = AddNewBiasNode(bias_data, func_graph, kernel_num, weight_tensor->data_type()); in GenNewConvTensor()
246 delete[] bias_data; in GenNewConvTensor()
[all …]
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dscale_int8.cc32 int8_t *scale_data, const std::vector<int> &bias_shape, int8_t *bias_data,
66 … int8_t *scale_data, const std::vector<int> &bias_shape, int8_t *bias_data, in Prepare() argument
83 bias_tensor_.set_data(bias_data); in Prepare()
107 int8_t bias_data[4] = {3, 3, 3, 3}; in TEST_F() local
120 …Prepare(input_shape, input_data, scale_shape, scale_data, bias_shape, bias_data, output_shape, out… in TEST_F()
133 int8_t bias_data[4] = {3, 3, 3, 3}; in TEST_F() local
146 …Prepare(input_shape, input_data, scale_shape, scale_data, bias_shape, bias_data, output_shape, out… in TEST_F()
159 int8_t bias_data[4] = {3, 3, 3, 3}; in TEST_F() local
172 …Prepare(input_shape, input_data, scale_shape, scale_data, bias_shape, bias_data, output_shape, out… in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dlstm_fp16.cc117 auto bias_data = bias->data(); in InitInputWeightBias() local
118 CHECK_NULL_RETURN(bias_data); in InitInputWeightBias()
127 …PackLstmBiasFp32ToFp16(input_bias_, reinterpret_cast<float *>(bias_data), weight_batch_, lstm_para… in InitInputWeightBias()
130 …PackLstmBiasFp16(input_bias_, reinterpret_cast<float16_t *>(bias_data), weight_batch_, lstm_param_… in InitInputWeightBias()
178 auto bias_data = bias->data(); in InitStateWeightBias() local
179 CHECK_NULL_RETURN(bias_data); in InitStateWeightBias()
188 …auto state_bias_data = reinterpret_cast<float *>(bias_data) + gate_num * lstm_param_->hidden_size_; in InitStateWeightBias()
192 …auto state_bias_data = reinterpret_cast<float16_t *>(bias_data) + gate_num * lstm_param_->hidden_s… in InitStateWeightBias()
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv1x1_init_int8_wrapper.c23 int8_t **packed_weight, int32_t **bias_data) { in Conv1x1Init() argument
24 if (packed_weight == NULL || bias_data == NULL) { in Conv1x1Init()
88 *bias_data = bias_data_; in Conv1x1Init()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dlstm_fp32_tests.cc73 …std::vector<float> bias_data = {-0.00207639, 0.16391152, -0.00069344, -0.32945693, -0.367423, 0… in InitLstmForwardCreator() local
82 memcpy(bias->MutableData(), bias_data.data(), bias_data.size() * sizeof(float)); in InitLstmForwardCreator()
231 std::vector<float> bias_data = { in InitLstmBackwardCreator() local
243 memcpy(bias->MutableData(), bias_data.data(), bias_data.size() * sizeof(float)); in InitLstmBackwardCreator()

123