Searched refs:bias_data (Results 1 – 6 of 6) sorted by relevance
/frameworks/ml/nn/common/operations/internal/reference/ |
D | depthwiseconv_float.h | 30 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()
|
D | depthwiseconv_uint8.h | 33 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()
|
D | reference_ops.h | 162 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/ |
D | optimized_ops.h | 223 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 …]
|
D | depthwiseconv_uint8.h | 1273 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 …]
|
D | depthwiseconv_float.h | 605 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,
|