Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_ops_test.cc43 const Tensor& bias_data, Tensor* out)>;
133 const Tensor& bias_data, Tensor* output, in RunConv2DWithBias() argument
145 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBias()
152 const Tensor& bias_data, Tensor* output, in RunConv2DWithBiasAndRelu() argument
164 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBiasAndRelu()
223 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias() argument
224 RunConv2DWithBias(input_data, filter_data, bias_data, out); in VerifyConv2DWithBias()
229 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias() argument
230 RunMklFusedConv2DOp(input_data, filter_data, {bias_data}, {"BiasAdd"}, in VerifyConv2DWithBias()
248 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu() argument
[all …]
Dconv_ops_test.cc540 const Tensor& bias_data, Tensor* out)>;
610 const Tensor& bias_data, Tensor* output, in RunConv2DWithBias() argument
622 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBias()
629 const Tensor& bias_data, Tensor* output, in RunConv2DWithBiasAndRelu() argument
641 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunConv2DWithBiasAndRelu()
842 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBias() argument
843 RunConv2DWithBias(input_data, filter_data, bias_data, out); in VerifyConv2DWithBias()
848 const Tensor& bias_data, in VerifyConv2DWithBias()
850 RunFusedConv2DOp(input_data, filter_data, {bias_data}, {"BiasAdd"}, out); in VerifyConv2DWithBias()
867 const Tensor& bias_data, Tensor* out) { in VerifyConv2DWithBiasAndRelu() argument
[all …]
Dconv_ops_fused_impl.h159 explicit BiasAddOutputKernel(const T* bias_data) : bias_data(bias_data) {} in BiasAddOutputKernel()
168 const T* bias_base = bias_data + i; in operator()
180 const T* bias_data;
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h33 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
56 if (bias_data) { in FullyConnected()
57 bias_value = bias_data[out_c]; in FullyConnected()
69 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
102 if (bias_data) { in FullyConnected()
103 acc += bias_data[out_c]; in FullyConnected()
118 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
146 int32 accum = bias_data[out_c]; in FullyConnected()
172 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected() argument
252 int32 acc = accum[i] + bias_data[c + i]; in ShuffledFullyConnected()
[all …]
Ddepthwiseconv_uint8.h97 const RuntimeShape& bias_shape, const int32* bias_data, in Run()
158 if (bias_data) { in Run()
159 acc += bias_data[oc]; in Run()
182 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument
188 bias_data, output_shape, in DepthwiseConv()
Dlegacy_reference_ops.h41 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
63 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
68 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
74 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
84 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
90 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
100 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
103 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
111 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
140 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
[all …]
Ddepthwiseconv_float.h29 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument
83 if (bias_data) { in DepthwiseConv()
84 bias_value = bias_data[oc]; in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc109 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv() argument
142 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
166 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
200 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
217 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
236 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
245 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
274 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
280 bias_shape, bias_data, output_shape, output_data); in DispatchDepthwiseConv()
292 std::int32_t filter_offset, const std::int32_t* bias_data, in TestOneDepthwiseConvWithGivenOutputShift() argument
[all …]
Ddepthwiseconv_float_test.cc36 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv() argument
41 filter_data, bias_shape, bias_data, output_shape, in TestOneDepthwiseConv()
44 filter_data, bias_shape, bias_data, output_shape, in TestOneDepthwiseConv()
119 std::vector<float> bias_data(output_depth); in TryTestOneDepthwiseConv() local
126 FillRandom(&bias_data, -bias_amplitude, bias_amplitude); in TryTestOneDepthwiseConv()
140 bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
162 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
167 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
173 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
183 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
189 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
199 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
202 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
210 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
239 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
[all …]
Ddepthwiseconv_uint8_3x3_filter.h3281 const uint8* filter_data, const int32* bias_data,
3306 const int32* bias_ptr = bias_data;
3363 ConvKernel::Run(input_data, filter_data, bias_data, output_data, 0,
3379 const int32* bias_data,
3386 input_data, filter_ptr, bias_data, output_data, &params);
3402 input_ptr, filter_ptr, bias_data, output_ptr, &params);
3411 input_ptr, filter_ptr, bias_data, output_ptr, &params);
3417 input_ptr, filter_ptr, bias_data, output_ptr, &params);
3427 input_ptr, filter_ptr, bias_data, output_ptr, &params);
3442 input_ptr, filter_ptr, bias_data, output_ptr, &params);
[all …]
Ddepthwiseconv_uint8_transitional.h114 const int32* bias_data,
124 adjusted_bias_block[s][z] = bias_data[4 * s + z];
133 static void Run(const uint8* filter_data, const int32* bias_data,
148 bias_data + 2 * bias_increment * j_depth, adjusted_bias_block);
162 static inline void Run(const uint8* filter_data, const int32* bias_data,
247 adjusted_bias_data_0[z] = bias_data[z];
248 adjusted_bias_data_1[z] = bias_data[4 + z];
267 bias_data += 2 * bias_increment;
278 const uint8* filter_data, const int32* bias_data,
363 int32x4_t adjusted_bias_data_a = vld1q_s32(bias_data);
[all …]
Ddepthwiseconv_uint8.h1589 const int32* bias_data,
1594 const int32x4_t b = vdupq_n_s32(bias_data[0]);
1605 int32x4_t b = vdupq_n_s32(bias_data[0]);
1606 b = vsetq_lane_s32(bias_data[1], b, 1);
1607 b = vsetq_lane_s32(bias_data[1], b, 3);
1618 const int32x4_t b = vld1q_s32(bias_data);
1629 const int32x4_t b0 = vld1q_s32(bias_data);
1630 const int32x4_t b1 = vld1q_s32(bias_data + 4);
1642 const int32x4_t b0 = vld1q_s32(bias_data);
1643 const int32x4_t b1 = vld1q_s32(bias_data + 4);
[all …]
Dmultithreaded_conv.h139 const float* bias_data, const RuntimeShape& output_shape, in Conv() argument
170 output_activation_min, output_activation_max, bias_shape, bias_data, in Conv()
Doptimized_ops.h190 const float* bias_data, in AddBiasAndEvalActivationFunction() argument
205 auto b0 = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction()
206 auto b1 = vld1q_f32(bias_data + i + 4); in AddBiasAndEvalActivationFunction()
207 auto b2 = vld1q_f32(bias_data + i + 8); in AddBiasAndEvalActivationFunction()
208 auto b3 = vld1q_f32(bias_data + i + 12); in AddBiasAndEvalActivationFunction()
231 auto b = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction()
239 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i], in AddBiasAndEvalActivationFunction()
253 array_data[array_offset + i] + bias_data[i], output_activation_min, in AddBiasAndEvalActivationFunction()
298 const int32* bias_data, int32 accum_multiplier, in GEMVForLstmCell() argument
318 const int32* bias_ptr = bias_data; in GEMVForLstmCell()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h29 const int32* bias_data, const RuntimeShape& output_shape, in ConvPerChannel() argument
53 if (bias_data) { in ConvPerChannel()
109 if (bias_data) { in ConvPerChannel()
110 acc += bias_data[out_channel]; in ConvPerChannel()
Dfully_connected.h27 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
54 if (bias_data) { in FullyConnected()
55 acc += bias_data[out_c]; in FullyConnected()
Ddepthwise_conv.h27 const int32* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel() argument
105 if (bias_data) { in DepthwiseConvPerChannel()
106 acc += bias_data[output_channel]; in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc47 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine() local
75 float& bias_val = bias_data[i]; in FuseAddOrSubParamsIntoPrecedingAffine()
118 float* const bias_data = bias_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
154 bias_data[c] *= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
156 bias_data[c] /= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
Dfuse_binary_into_following_affine.cc75 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() local
92 bias_data[d] += accumulation; in FuseAddOrSubParamsIntoFollowingAffine()
106 bias_data[c] += accumulation; in FuseAddOrSubParamsIntoFollowingAffine()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h285 const float* bias_data, in AddBiasAndEvalActivationFunction() argument
300 auto b0 = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction()
301 auto b1 = vld1q_f32(bias_data + i + 4); in AddBiasAndEvalActivationFunction()
302 auto b2 = vld1q_f32(bias_data + i + 8); in AddBiasAndEvalActivationFunction()
303 auto b3 = vld1q_f32(bias_data + i + 12); in AddBiasAndEvalActivationFunction()
326 auto b = vld1q_f32(bias_data + i); in AddBiasAndEvalActivationFunction()
334 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i], in AddBiasAndEvalActivationFunction()
348 array_data[array_offset + i] + bias_data[i], output_activation_min, in AddBiasAndEvalActivationFunction()
357 inline void AddBiasAndEvalActivationFunction(const float* bias_data, in AddBiasAndEvalActivationFunction() argument
364 DimsToShape(bias_dims), bias_data, in AddBiasAndEvalActivationFunction()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv_test.cc32 std::initializer_list<float> bias_data, in TestDepthwiseConvFloat() argument
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"), in TestDepthwiseConvFloat()
114 std::initializer_list<int32_t> bias_data, float bias_min, float bias_max, in TestDepthwiseConvQuantized() argument
132 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_min, in TestDepthwiseConvQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Dfully_connected_test.cc604 std::vector<float> bias_data(output_depth); in SimpleTestQuantizedInt16OutputCase() local
608 for (auto& b : bias_data) { in SimpleTestQuantizedInt16OutputCase()
612 m.SetBias(bias_data); in SimpleTestQuantizedInt16OutputCase()
620 float accum = bias_data[o]; in SimpleTestQuantizedInt16OutputCase()
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h541 const float* bias_data,
547 memcpy(acc_buffer + i * output_depth, bias_data,
555 const float* bias_data, const Dims<4>& bias_dims, int stride,
645 DepthwiseConvInitAccBuffer(num_output_pixels, output_depth, bias_data,
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc99 const RuntimeShape& bias_shape, const int32* bias_data, in DepthwiseConvOptimizedForFilterWidthEight() argument
255 if (bias_data) { in DepthwiseConvOptimizedForFilterWidthEight()
256 acc += bias_data[oc]; in DepthwiseConvOptimizedForFilterWidthEight()

12