| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| depthwiseconv_float_test.cc | 36 const float* bias_data, const RuntimeShape& output_shape) {
41 filter_data, bias_shape, bias_data, output_shape,
44 filter_data, bias_shape, bias_data, output_shape,
119 std::vector<float> bias_data(output_depth);
126 FillRandom(&bias_data, -bias_amplitude, bias_amplitude);
140 bias_shape_inference, bias_data.data(),
|
| depthwiseconv_quantized_test.cc | 109 const RuntimeShape& bias_shape, const int32* bias_data,
142 bias_shape, bias_data, output_shape, output_data);
166 bias_shape, bias_data, output_shape, output_data);
200 bias_shape, bias_data, output_shape, output_data);
217 bias_shape, bias_data, output_shape, output_data);
236 bias_shape, bias_data, output_shape, output_data);
245 bias_shape, bias_data, output_shape, output_data);
274 bias_shape, bias_data, output_shape, output_data);
280 bias_shape, bias_data, output_shape, output_data);
292 std::int32_t filter_offset, const std::int32_t* bias_data,
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| fully_connected.h | 27 const int32* bias_data, const RuntimeShape& output_shape,
54 if (bias_data) {
55 acc += bias_data[out_c];
|
| conv.h | 29 const int32* bias_data, const RuntimeShape& output_shape,
53 if (bias_data) {
109 if (bias_data) {
110 acc += bias_data[out_channel];
|
| depthwise_conv.h | 27 const int32* bias_data, const RuntimeShape& output_shape,
105 if (bias_data) {
106 acc += bias_data[output_channel];
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| fully_connected.h | 33 const float* bias_data, const RuntimeShape& output_shape,
56 if (bias_data) {
57 bias_value = bias_data[out_c];
69 const int32* bias_data, const RuntimeShape& output_shape,
102 if (bias_data) {
103 acc += bias_data[out_c];
118 const int32* bias_data, const RuntimeShape& output_shape,
146 int32 accum = bias_data[out_c];
172 const int32* bias_data, const RuntimeShape& output_shape,
252 int32 acc = accum[i] + bias_data[c + i]
[all...] |
| depthwiseconv_float.h | 29 const float* bias_data, const RuntimeShape& output_shape,
83 if (bias_data) {
84 bias_value = bias_data[oc];
|
| depthwiseconv_uint8.h | 97 const RuntimeShape& bias_shape, const int32* bias_data,
158 if (bias_data) {
159 acc += bias_data[oc];
182 const int32* bias_data, const RuntimeShape& output_shape,
188 bias_data, output_shape,
|
| legacy_reference_ops.h | 41 const float* bias_data, const Dims<4>& bias_dims,
63 bias_data, DimsToShape(output_dims), output_data);
68 const float* bias_data, const Dims<4>& bias_dims,
74 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
84 const float* bias_data, const Dims<4>& bias_dims,
90 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
100 const float* bias_data, const Dims<4>& bias_dims, int stride,
103 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data,
111 const int32* bias_data, const Dims<4>& bias_dims,
140 bias_data, DimsToShape(output_dims), output_data)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_fused_ops_test.cc | 43 const Tensor& bias_data, Tensor* out)>;
133 const Tensor& bias_data, Tensor* output,
145 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data)));
152 const Tensor& bias_data, Tensor* output,
164 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data)));
223 const Tensor& bias_data, Tensor* out) {
224 RunConv2DWithBias(input_data, filter_data, bias_data, out);
229 const Tensor& bias_data, Tensor* out) {
230 RunMklFusedConv2DOp(input_data, filter_data, {bias_data}, {"BiasAdd"},
248 const Tensor& bias_data, Tensor* out)
[all...] |
| conv_ops_test.cc | 540 const Tensor& bias_data, Tensor* out)>;
610 const Tensor& bias_data, Tensor* output,
622 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data)));
629 const Tensor& bias_data, Tensor* output,
641 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data)));
[all...] |
| conv_ops_fused_impl.h | 159 explicit BiasAddOutputKernel(const T* bias_data) : bias_data(bias_data) {}
168 const T* bias_base = bias_data + i;
180 const T* bias_data; member in struct:tensorflow::BiasAddOutputKernel
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| legacy_optimized_ops.h | 140 const float* bias_data, const Dims<4>& bias_dims,
162 bias_data, DimsToShape(output_dims), output_data);
167 const float* bias_data, const Dims<4>& bias_dims,
173 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
183 const float* bias_data, const Dims<4>& bias_dims,
189 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data,
199 const float* bias_data, const Dims<4>& bias_dims, int stride,
202 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data,
210 const int32* bias_data, const Dims<4>& bias_dims,
239 bias_data, DimsToShape(output_dims), output_data)
[all...] |
| depthwiseconv_uint8_3x3_filter.h | [all...] |
| depthwiseconv_uint8_transitional.h | 114 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...] |
| multithreaded_conv.h | 139 const float* bias_data, const RuntimeShape& output_shape,
170 output_activation_min, output_activation_max, bias_shape, bias_data,
|
| depthwiseconv_uint8.h | [all...] |
| optimized_ops.h | 190 const float* bias_data,
205 auto b0 = vld1q_f32(bias_data + i);
206 auto b1 = vld1q_f32(bias_data + i + 4);
207 auto b2 = vld1q_f32(bias_data + i + 8);
208 auto b3 = vld1q_f32(bias_data + i + 12);
231 auto b = vld1q_f32(bias_data + i);
239 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i],
253 array_data[array_offset + i] + bias_data[i], output_activation_min,
298 const int32* bias_data, int32 accum_multiplier,
318 const int32* bias_ptr = bias_data;
[all...] |
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| fuse_binary_into_preceding_affine.cc | 47 float* const bias_data = bias_buffer.data.data(); local
75 float& bias_val = bias_data[i];
118 float* const bias_data = bias_buffer.data.data(); local
154 bias_data[c] *= operand_data[operand_channel];
156 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/lite/experimental/micro/kernels/ |
| depthwise_conv_test.cc | 32 std::initializer_list<float> bias_data,
49 CreateFloatTensor(bias_data, bias_dims, "bias_tensor"),
114 std::initializer_list<int32_t> bias_data, float bias_min, float bias_max,
132 CreateQuantized32Tensor(bias_data, bias_dims, "bias_tensor", bias_min,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 285 const float* bias_data,
300 auto b0 = vld1q_f32(bias_data + i);
301 auto b1 = vld1q_f32(bias_data + i + 4);
302 auto b2 = vld1q_f32(bias_data + i + 8);
303 auto b3 = vld1q_f32(bias_data + i + 12);
326 auto b = vld1q_f32(bias_data + i);
334 array_ptr[i] = ActivationFunctionWithMinMax(array_ptr[i] + bias_data[i],
348 array_data[array_offset + i] + bias_data[i], output_activation_min,
357 inline void AddBiasAndEvalActivationFunction(const float* bias_data,
364 DimsToShape(bias_dims), bias_data,
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/ |
| fully_connected_test.cc | 604 std::vector<float> bias_data(output_depth);
608 for (auto& b : bias_data) {
612 m.SetBias(bias_data);
620 float accum = bias_data[o];
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/ |
| depthwise_conv.cc | 99 const RuntimeShape& bias_shape, const int32* bias_data,
255 if (bias_data) {
256 acc += bias_data[oc];
|
| /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,
|
