| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| depthwiseconv_float.h | 27 const float* bias_data, const Dims<4>& bias_dims,
70 if (bias_data) {
71 bias_value = bias_data[Offset(bias_dims, oc, 0, 0, 0)];
88 const float* bias_data, const Dims<4>& bias_dims,
94 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
104 const float* bias_data, const Dims<4>& bias_dims, int stride,
107 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data,
|
| depthwiseconv_uint8.h | 32 const int32* bias_data, const Dims<4>& bias_dims,
76 if (bias_data) {
77 acc += bias_data[Offset(bias_dims, oc, 0, 0, 0)];
98 const int32* bias_data, const Dims<4>& bias_dims,
109 filter_offset, bias_data, bias_dims, stride_width,
121 const int32* bias_data, const Dims<4>& bias_dims, int stride,
128 filter_dims, filter_offset, bias_data, bias_dims, stride,
|
| reference_ops.h | 159 const float* bias_data, const Dims<4>& bias_dims,
170 if (bias_data) {
206 if (bias_data) {
207 bias_value = bias_data[Offset(bias_dims, out_channel, 0, 0, 0)];
223 const float* bias_data, const Dims<4>& bias_dims, int stride_width,
229 Conv(input_data, input_dims, filter_data, filter_dims, bias_data, bias_dims,
239 const float* bias_data, const Dims<4>& bias_dims, int stride,
243 Conv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data,
251 const int32* bias_data, const Dims<4>& bias_dims,
300 if (bias_data) {
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| cblas_conv.h | 36 const float* bias_data, const Dims<4>& bias_dims,
85 bias_data, bias_dims, output_data, output_dims, output_activation_min,
|
| depthwiseconv_float.h | [all...] |
| multithreaded_conv.h | 166 const float* bias_data, const Dims<4>& bias_dims,
188 bias_data, bias_dims, output_data, output_dims, output_activation_min,
|
| depthwiseconv_uint8.h | [all...] |
| optimized_ops.h | 229 inline void AddBiasAndEvalActivationFunction(const float* bias_data,
247 auto b0 = vld1q_f32(bias_data + i);
248 auto b1 = vld1q_f32(bias_data + i + 4);
249 auto b2 = vld1q_f32(bias_data + i + 8);
250 auto b3 = vld1q_f32(bias_data + i + 12);
273 auto b = vld1q_f32(bias_data + i);
281 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i],
295 array_data[array_offset + i] + bias_data[i], output_activation_min,
304 void AddBiasAndEvalActivationFunction(const float* bias_data,
310 AddBiasAndEvalActivationFunction(bias_data, bias_dims, array_data, array_dims
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| fuse_binary_into_preceding_affine.cc | 47 float* const bias_data = bias_buffer.data.data(); local
64 float& bias_val = bias_data[i];
106 float* const bias_data = bias_buffer.data.data(); local
142 bias_data[c] *= operand_data[operand_channel];
144 bias_data[c] /= operand_data[operand_channel];
|
| fuse_binary_into_following_affine.cc | 75 float* const bias_data = bias_buffer.data.data(); local
92 bias_data[d] += accumulation;
106 bias_data[c] += accumulation;
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| depthwiseconv_float.h | 541 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/contrib/lite/toco/ |
| export_tensorflow.cc | 379 const float* bias_data = local
381 ConvertFloatTensorConst(bias_array_name, bias_shape_1d, bias_data,
458 const float* bias_data = local
460 ConvertFloatTensorConst(bias_name, bias_shape_1d, bias_data,
560 const float* bias_data = local
563 bias_shape_1d, bias_data, AxesOrder::kOneAxis, local
1274 const float* bias_data = local
[all...] |
