Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/internal/reference/
Ddepthwiseconv_float.h30 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
71 if (bias_data) { in DepthwiseConv()
72 bias_value = bias_data[Offset(bias_dims, oc, 0, 0, 0)]; in DepthwiseConv()
Ddepthwiseconv_uint8.h33 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
87 if (bias_data) { in DepthwiseConv()
88 acc += bias_data[Offset(bias_dims, oc, 0, 0, 0)]; in DepthwiseConv()
Dreference_ops.h162 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv() argument
171 if (bias_data) { in Conv()
207 if (bias_data) { in Conv()
208 bias_value = bias_data[Offset(bias_dims, out_channel, 0, 0, 0)]; in Conv()
222 const int32* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv() argument
278 if (bias_data) { in Conv()
279 acc += bias_data[Offset(bias_dims, out_channel, 0, 0, 0)]; in Conv()
379 const float* bias_data, const Dims<4>& bias_dims, in FullyConnected() argument
400 if (bias_data) { in FullyConnected()
401 bias_value = bias_data[Offset(bias_dims, out_c, 0, 0, 0)]; in FullyConnected()
[all …]
/frameworks/ml/nn/common/operations/internal/optimized/
Doptimized_ops.h223 void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction() argument
240 auto b0 = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction()
241 auto b1 = vld1q_f32(bias_data + i + 4); in AddBiasAndEvalActivationFunction()
242 auto b2 = vld1q_f32(bias_data + i + 8); in AddBiasAndEvalActivationFunction()
243 auto b3 = vld1q_f32(bias_data + i + 12); in AddBiasAndEvalActivationFunction()
280 auto b = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction()
296 array_ptr[i] = ActivationFunction<Ac>(array_ptr[i] + bias_data[i]); in AddBiasAndEvalActivationFunction()
302 void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction() argument
314 ActivationFunction<Ac>(array_data[array_offset + i] + bias_data[i]); in AddBiasAndEvalActivationFunction()
335 const float* bias_data, const Dims<4>& bias_dims, in FullyConnected() argument
[all …]
Ddepthwiseconv_uint8.h1273 const int32* bias_data,
1278 const int32x4_t b = vdupq_n_s32(bias_data[0]);
1289 int32x4_t b = vdupq_n_s32(bias_data[0]);
1290 b = vsetq_lane_s32(bias_data[1], b, 1);
1291 b = vsetq_lane_s32(bias_data[1], b, 3);
1302 const int32x4_t b = vld1q_s32(bias_data);
1313 const int32x4_t b0 = vld1q_s32(bias_data);
1314 const int32x4_t b1 = vld1q_s32(bias_data + 4);
1326 const int32x4_t b0 = vld1q_s32(bias_data);
1327 const int32x4_t b1 = vld1q_s32(bias_data + 4);
[all …]
Ddepthwiseconv_float.h605 const float* bias_data,
608 memcpy(acc_buffer + i * output_depth, bias_data,
616 const float* bias_data, const Dims<4>& bias_dims,
708 DepthwiseConvInitAccBuffer(num_output_pixels, output_depth, bias_data,