Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h34 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight() argument
69 float bias_value = bias_data[i]; in FullyConnectedSparseWeight()
80 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl() argument
108 float bias_value = bias_data[i]; in FullyConnectedSparseWeight1x4Impl()
120 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Task()
130 bias_data(bias_data), in FullyConnectedSparseWeight1x4Task()
140 bias_shape, bias_data, output_shape, output_data, thread_start, in Run()
152 const float* bias_data; member
168 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4() argument
181 bias_shape, bias_data, output_shape, output_data, 0, batches, in FullyConnectedSparseWeight1x4()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc127 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvGeneral() argument
135 bias_data, output_shape, output_data, thread_start, thread_end, in DispatchDepthwiseConvGeneral()
144 const int32* bias_data, const std::int32_t* output_shift_adjust, in DispatchDepthwiseConvGeneral() argument
150 input_data, filter_shape, filter_data, bias_shape, bias_data, in DispatchDepthwiseConvGeneral()
163 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl() argument
201 bias_shape, bias_data, output_shape, output_data, in DispatchDepthwiseConvImpl()
209 bias_shape, bias_data, output_shape, output_data, in DispatchDepthwiseConvImpl()
236 bias_shape, bias_data, output_shape, output_data, /*thread_start=*/0, in DispatchDepthwiseConvImpl()
272 bias_shape, bias_data, output_shape, output_data, /*thread_start=*/0, in DispatchDepthwiseConvImpl()
291 bias_shape, bias_data, output_shape, output_data, /*thread_start=*/0, in DispatchDepthwiseConvImpl()
[all …]
Ddepthwiseconv_per_channel_quantized_test.cc44 const int32* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier() argument
94 if (bias_data) { in PickOutputMultiplier()
95 acc += bias_data[output_channel]; in PickOutputMultiplier()
119 const RuntimeShape& bias_shape_inference, const std::int32_t* bias_data, in PickReasonableMultiplier() argument
126 bias_shape_inference, bias_data, output_shape_inference, in PickReasonableMultiplier()
264 std::vector<std::int32_t> bias_data(output_depth); in TryTestOneDepthwiseConv3x3Filter() local
268 FillRandom(&bias_data, -1000, 1000); in TryTestOneDepthwiseConv3x3Filter()
298 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
314 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
320 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
[all …]
Ddepthwiseconv_float_test.cc37 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv() argument
42 filter_data, bias_shape, bias_data, output_shape, in TestOneDepthwiseConv()
46 bias_data, output_shape, output_data.data(), CpuFlags(), in TestOneDepthwiseConv()
122 std::vector<float> bias_data(output_depth); in TryTestOneDepthwiseConv() local
129 FillRandom(&bias_data, -bias_amplitude, bias_amplitude); in TryTestOneDepthwiseConv()
143 bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv()
Ddepthwiseconv_per_channel_quantized_16x8_test.cc43 const std::int64_t* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier() argument
93 if (bias_data) { in PickOutputMultiplier()
94 acc += bias_data[output_channel]; in PickOutputMultiplier()
118 const RuntimeShape& bias_shape_inference, const std::int64_t* bias_data, in PickReasonableMultiplier() argument
125 bias_shape_inference, bias_data, output_shape_inference, in PickReasonableMultiplier()
219 std::vector<std::int64_t> bias_data(output_depth); in TryTestOneDepthwiseConv3x3Filter() local
224 bias_data.data()[i] = 0; in TryTestOneDepthwiseConv3x3Filter()
255 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
264 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
276 IntToFloat(&bias_data_float, &bias_data); in TryTestOneDepthwiseConv3x3Filter()
Dconv_per_channel_quantized_16x8_test.cc43 const std::int64_t* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier() argument
95 if (bias_data) { in PickOutputMultiplier()
96 acc += bias_data[output_channel]; in PickOutputMultiplier()
119 const RuntimeShape& bias_shape_inference, const std::int64_t* bias_data, in PickReasonableMultiplier() argument
126 bias_shape_inference, bias_data, output_shape_inference, in PickReasonableMultiplier()
220 std::vector<std::int64_t> bias_data(output_depth); in TryTestOneConvFilter() local
231 bias_data.data()[i] = 0; in TryTestOneConvFilter()
261 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneConvFilter()
271 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneConvFilter()
283 IntToFloat(&bias_data_float, &bias_data); in TryTestOneConvFilter()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h45 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConvHybridGeneral() argument
222 const float32x4_t bias_32x4 = vld1q_f32(bias_data + c); in DepthwiseConvHybridGeneral()
242 float_acc += bias_data[c]; in DepthwiseConvHybridGeneral()
263 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding() argument
302 filter_shape, filter_data, bias_shape, bias_data, output_shape, in DepthwiseConvHybridWithRounding()
313 filter_shape, filter_data, bias_shape, bias_data, output_shape, in DepthwiseConvHybridWithRounding()
322 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl() argument
329 filter_shape, filter_data, bias_shape, bias_data, output_shape, in DepthwiseConvHybridImpl()
343 const TS* bias_data, in DepthwiseConvHybridWorkerTask()
357 bias_data(bias_data), in DepthwiseConvHybridWorkerTask()
[all …]
Ddepthwise_conv_hybrid_3x3_filter.h2827 const float* bias_data, float* output_data,
2854 const float* bias_ptr = bias_data;
2918 bias_data, output_data, 0, params.output_depth,
2935 const int8* filter_data, const float* bias_data, float* output_data,
2942 filter_ptr, bias_data, output_data,
2960 input_scale, input_ptr, filter_ptr, bias_data,
2971 input_scale, input_ptr, filter_ptr, bias_data, output_ptr,
2979 input_scale, input_ptr, filter_ptr, bias_data, output_ptr,
2991 input_scale, input_ptr, filter_ptr, bias_data, output_ptr,
3008 input_scale, input_ptr, filter_ptr, bias_data, output_ptr,
[all …]
Ddepthwise_conv_3x3_filter.h2661 const int32* bias_data, int8* output_data,
2686 const int32* bias_ptr = bias_data;
2750 bias_data, output_data, 0, params.output_depth,
2767 const int8* input_data, const int8* filter_data, const int32* bias_data,
2775 filter_ptr, bias_data, output_data,
2792 output_multiplier_ptr, output_shift_ptr, input_ptr, filter_ptr, bias_data,
2804 filter_ptr, bias_data, output_ptr,
2811 output_multiplier_ptr, output_shift_ptr, input_ptr, filter_ptr, bias_data,
2824 filter_ptr, bias_data, output_ptr,
2842 filter_ptr, bias_data, output_ptr,
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h28 const float* bias_data, const RuntimeShape& output_shape, in Conv() argument
48 if (bias_data) { in Conv()
88 if (bias_data) { in Conv()
89 bias_value = bias_data[out_channel]; in Conv()
104 const int32_t* bias_data, const RuntimeShape& output_shape, in Conv() argument
131 if (bias_data) { in Conv()
171 if (bias_data) { in Conv()
172 acc += bias_data[out_channel]; in Conv()
191 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel() argument
211 if (bias_data) { in HybridConvPerChannel()
[all …]
Dfully_connected.h30 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
53 if (bias_data) { in FullyConnected()
54 bias_value = bias_data[out_c]; in FullyConnected()
66 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
98 if (bias_data) { in FullyConnected()
99 acc += bias_data[out_c]; in FullyConnected()
114 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
141 int32_t accum = bias_data[out_c]; in FullyConnected()
168 const int32_t* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected() argument
246 int32_t acc = accum[i] + bias_data[c + i]; in ShuffledFullyConnected()
[all …]
Dtranspose_conv.h29 const float* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
50 if (bias_data) { in TransposeConv()
100 if (bias_data) { in TransposeConv()
106 out_channel)] += bias_data[out_channel]; in TransposeConv()
118 const int32_t* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
148 if (bias_data) { in TransposeConv()
198 if (bias_data) { in TransposeConv()
199 acc += bias_data[out_channel]; in TransposeConv()
Ddepthwiseconv_uint8.h96 const int32_t* bias_data, const RuntimeShape& output_shape, in Run()
157 if (bias_data) { in Run()
158 acc += bias_data[oc]; in Run()
180 const int32_t* bias_data, const RuntimeShape& output_shape, in RunPerChannel()
259 if (bias_data) { in RunPerChannel()
260 acc += bias_data[output_channel]; in RunPerChannel()
284 const int32_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument
290 bias_data, output_shape, in DepthwiseConv()
Dsvdf.h71 const int32_t* bias_data, int16_t* state_data, in EvalIntegerSVDF() argument
135 if (bias_data) { in EvalIntegerSVDF()
136 tensor_utils::VectorBatchVectorAdd(bias_data, n_unit, n_batch, in EvalIntegerSVDF()
157 const float* bias_data, float* scratch_data, float* state_data, in EvalFloatSVDF() argument
188 bias_data, params->activation, state_data, scratch_data, output_data); in EvalFloatSVDF()
196 const RuntimeShape& bias_shape, const float* bias_data, float* scratch, in EvalHybridSVDF() argument
243 bias_data, params->activation, state, scratch, output_data); in EvalHybridSVDF()
Dconv3d.h27 const float* bias_data, const RuntimeShape& output_shape, in Conv3D() argument
36 if (bias_data) { in Conv3D()
96 if (bias_data) { in Conv3D()
97 bias_value = bias_data[out_channel]; in Conv3D()
Dlegacy_reference_ops.h44 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
66 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
71 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
77 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
87 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
93 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
103 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
106 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
114 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
143 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h27 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
53 if (bias_data) { in FullyConnected()
54 acc += bias_data[out_c]; in FullyConnected()
69 const int64_t* bias_data, const RuntimeShape& output_shape, in FullyConnected() argument
93 if (bias_data) { in FullyConnected()
94 acc += bias_data[out_c]; in FullyConnected()
Dconv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in ConvPerChannel() argument
53 if (bias_data) { in ConvPerChannel()
111 if (bias_data) { in ConvPerChannel()
112 acc += bias_data[out_channel]; in ConvPerChannel()
134 const std::int64_t* bias_data, const RuntimeShape& output_shape, in ConvPerChannel() argument
156 if (bias_data) { in ConvPerChannel()
203 if (bias_data) { in ConvPerChannel()
204 acc += bias_data[out_channel]; in ConvPerChannel()
Dtranspose_conv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
45 if (bias_data) { in TransposeConv()
106 if (bias_data) { in TransposeConv()
107 acc += bias_data[out_channel]; in TransposeConv()
128 const std::int64_t* bias_data, const RuntimeShape& output_shape, in TransposeConv() argument
144 if (bias_data) { in TransposeConv()
203 if (bias_data) { in TransposeConv()
204 acc += bias_data[out_channel]; in TransposeConv()
Ddepthwise_conv.h27 const int32_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel() argument
104 if (bias_data) { in DepthwiseConvPerChannel()
105 acc += bias_data[output_channel]; in DepthwiseConvPerChannel()
127 const std::int64_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel() argument
190 if (bias_data) { in DepthwiseConvPerChannel()
191 acc += bias_data[output_channel]; in DepthwiseConvPerChannel()
212 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel() argument
272 if (bias_data && output_channel < bias_depth) { in DepthwiseConvHybridPerChannel()
273 acc_float += bias_data[output_channel]; in DepthwiseConvHybridPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_3x3_filter.h5393 const uint8* filter_data, const int32* bias_data,
5418 const int32* bias_ptr = bias_data;
5475 ConvKernel::Run(input_data, filter_data, bias_data, output_data, 0,
5492 const int32* bias_data,
5499 input_data, filter_ptr, bias_data, output_data, &params);
5515 input_ptr, filter_ptr, bias_data, output_ptr, &params);
5524 input_ptr, filter_ptr, bias_data, output_ptr, &params);
5530 input_ptr, filter_ptr, bias_data, output_ptr, &params);
5540 input_ptr, filter_ptr, bias_data, output_ptr, &params);
5555 input_ptr, filter_ptr, bias_data, output_ptr, &params);
[all …]
Ddepthwiseconv_multithread.h39 const TS* bias_data, const RuntimeShape& output_shape, in DepthwiseConvWorkerTask()
48 bias_data_(bias_data), in DepthwiseConvWorkerTask()
123 const TS* bias_data, const RuntimeShape& output_shape, in DepthwiseConv() argument
151 filter_data, bias_shape, bias_data, output_shape, in DepthwiseConv()
175 filter_data, bias_shape, bias_data, output_shape, in DepthwiseConv()
Dlegacy_optimized_ops.h140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
165 DimsToShape(bias_dims), bias_data, output_shape, in DepthwiseConv()
172 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
178 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
188 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv() argument
194 DepthwiseConv(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
204 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv() argument
207 DepthwiseConv<Ac>(input_data, input_dims, filter_data, filter_dims, bias_data, in DepthwiseConv()
217 const int32* bias_data, const RuntimeShape& output_shape, in LegacyDepthwiseConvWithRounding() argument
254 bias_data, output_shape, output_data, thread_start, thread_end, in LegacyDepthwiseConvWithRounding()
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dfully_connected_tester.cc133 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
147 bias_data[oc] = value_rng(); in CreateTfLiteModel()
159 builder.CreateVector(reinterpret_cast<const uint8_t*>(bias_data.data()), in CreateTfLiteModel()
160 sizeof(uint16_t) * bias_data.size()))); in CreateTfLiteModel()
180 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
193 bias_data[oc] = value_rng(); in CreateTfLiteModel()
205 builder.CreateVector(reinterpret_cast<const uint8_t*>(bias_data.data()), in CreateTfLiteModel()
206 sizeof(float) * bias_data.size()))); in CreateTfLiteModel()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_test.cc37 const Tensor& bias_data, Tensor* out)>;
101 const Tensor& bias_data, bool transpose_a, in RunMatMulWithBias() argument
114 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunMatMulWithBias()
120 const Tensor& lhs_data, const Tensor& rhs_data, const Tensor& bias_data, in RunMatMulWithBiasAndActivation() argument
133 ops::Const(root.WithOpName("bias"), Input::Initializer(bias_data))); in RunMatMulWithBiasAndActivation()
225 const Tensor& bias_data, Tensor* out) { in VerifyMatMulWithBias() argument
226 RunMatMulWithBias(input_data, filter_data, bias_data, transpose_a, in VerifyMatMulWithBias()
232 const Tensor& bias_data, Tensor* out) { in VerifyMatMulWithBias() argument
233 RunFusedMatMulOp(input_data, filter_data, {bias_data}, {"BiasAdd"}, in VerifyMatMulWithBias()
247 const Tensor& bias_data, in VerifyConv2DWithBiasAndActivation()
[all …]

1234