Home
last modified time | relevance | path

Searched defs:bias_data (Results 1 – 25 of 93) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h37 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
83 const RuntimeShape& bias_shape, const int32_t* bias_data, in FullyConnectedSparseWeight1x16Impl()
119 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl()
191 const float* bias_data; member
203 const RuntimeShape& bias_shape, const int32_t* bias_data, in FullyConnectedSparseWeight1x16()
227 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Ddepthwiseconv_quantized_test.cc127 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvGeneral()
144 const int32* bias_data, const std::int32_t* output_shift_adjust, in DispatchDepthwiseConvGeneral()
163 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl()
366 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl()
533 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
618 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvWithGivenOutputShift()
755 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvBisectOutputShift()
814 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConv()
895 std::vector<std::int32_t> bias_data(output_depth); in TryTestDepthwiseConv() local
H A Dconv_per_channel_quantized_16x8_test.cc43 const std::int64_t* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier()
119 const RuntimeShape& bias_shape_inference, const std::int64_t* bias_data, in PickReasonableMultiplier()
220 std::vector<std::int64_t> bias_data(output_depth); in TryTestOneConvFilter() local
H A Ddepthwiseconv_per_channel_quantized_16x8_test.cc43 const std::int64_t* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier()
118 const RuntimeShape& bias_shape_inference, const std::int64_t* bias_data, in PickReasonableMultiplier()
219 std::vector<std::int64_t> bias_data(output_depth); in TryTestOneDepthwiseConv3x3Filter() local
H A Ddepthwiseconv_per_channel_quantized_test.cc44 const int32* bias_data, const RuntimeShape& output_shape, in PickOutputMultiplier()
119 const RuntimeShape& bias_shape_inference, const std::int32_t* bias_data, in PickReasonableMultiplier()
264 std::vector<std::int32_t> bias_data(output_depth); in TryTestOneDepthwiseConv3x3Filter() local
H A Ddepthwiseconv_float_test.cc37 const float* bias_data, const RuntimeShape& output_shape) { in TestOneDepthwiseConv()
122 std::vector<float> bias_data(output_depth); in TryTestOneDepthwiseConv() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlegacy_reference_ops.h46 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
73 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
89 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
105 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
116 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
151 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
171 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
194 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
210 const float* bias_data, const Dims<4>& bias_dims, in Conv()
236 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
[all …]
H A Dfully_connected.h33 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
69 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
117 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
171 const int32_t* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
H A Dsvdf.h71 const int32_t* bias_data, int16_t* state_data, in EvalIntegerSVDF()
157 const float* bias_data, float* scratch_data, float* state_data, in EvalFloatSVDF()
196 const RuntimeShape& bias_shape, const float* bias_data, float* scratch, in EvalHybridSVDF()
H A Dconv.h30 const float* bias_data, const RuntimeShape& output_shape, in Conv()
113 const int32_t* bias_data, const RuntimeShape& output_shape, in Conv()
207 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_ops_test.cc218 const Tensor& bias_data, in RunConv2DUnfused()
339 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedConv2D()
340 Tensor* out, const int padding) { in VerifyFusedConv2D()
536 const Tensor& bias_data, in RunDepthwiseConv2DUnfused()
630 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedDepthwiseConv2D()
631 Tensor* out, const int padding) { in VerifyFusedDepthwiseConv2D()
1295 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBias()
1313 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
1320 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
1332 const Tensor& bias_data, Tensor* output, in RunMklPadWithFusedConv2DAndBias()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h141 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
173 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
189 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
205 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
218 const int32* bias_data, const RuntimeShape& output_shape, in LegacyDepthwiseConvWithRounding()
272 const int32* bias_data, const RuntimeShape& output_shape, in LegacyDepthwiseConvImpl()
284 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
324 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
344 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
367 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op_test.cc101 const Tensor& bias_data, bool transpose_a, in RunMatMulWithBias()
120 const Tensor& lhs_data, const Tensor& rhs_data, const Tensor& bias_data, in RunMatMulWithBiasAndActivation()
225 const Tensor& bias_data, Tensor* out) { in VerifyMatMulWithBias()
232 const Tensor& bias_data, Tensor* out) { in VerifyMatMulWithBias()
257 const Tensor& bias_data, in VerifyConv2DWithBiasAndActivation()
258 Tensor* out) { in VerifyConv2DWithBiasAndActivation()
H A Dconv_ops_test.cc556 const Tensor& bias_data, const std::string& padding, in RunConv2DWithBias()
578 const Tensor& bias_data, const string& activation_type, in RunConv2DWithBiasAndActivation()
824 const Tensor& bias_data, Tensor* out) { in __anona74cd8200202()
832 const Tensor& bias_data, Tensor* out) { in __anona74cd8200302()
853 const Tensor& bias_data, Tensor* out) { in __anona74cd8200402()
864 Tensor* out) { in __anona74cd8200502()
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dquantized_linear_out.cpp50 const int32_t* __restrict__ bias_data = bias.const_data_ptr<int32_t>(); in _quantized_linear_asym8u() local
95 const int32_t* __restrict__ bias_data = bias.const_data_ptr<int32_t>(); in _quantized_linear_asym8s() local
140 const int32_t* __restrict__ bias_data = bias.const_data_ptr<int32_t>(); in _quantized_linear_per_tensor_asym8u() local
188 const int32_t* __restrict__ bias_data = bias.const_data_ptr<int32_t>(); in _quantized_linear_per_tensor_asym8s() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dfully_connected.h37 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnectedPerChannel()
79 const AccumScalar* bias_data, const RuntimeShape& output_shape, in FullyConnectedPerChannel()
117 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
161 const AccumScalar* bias_data, const RuntimeShape& output_shape, in FullyConnected()
H A Ddepthwise_conv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
129 const std::int64_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
214 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Ddepthwise_conv_hybrid.h47 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConvHybridGeneral()
265 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding()
324 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl()
384 const TS* bias_data; member
398 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
H A Dfully_connected.h36 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnectedPerChannel()
105 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DNormalization.cpp29 const float* bias_data, in compute_fused_params()
84 const float* bias_data = bias.template const_data_ptr<float>(); in q_batch_norm1d_impl() local
193 const float* bias_data = bias.template const_data_ptr<float>(); in q_batch_norm2d_impl() local
289 const float* bias_data = bias.template const_data_ptr<float>(); in q_batch_norm3d_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dtranspose_conv_tester.cc175 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
279 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
346 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
H A Dfully_connected_tester.cc143 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
195 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dunidirectional_sequence_rnn.mod.py21 recurrent_weights_data, bias_data, hidden_state_data, output_data): argument
107 bias_data = [ variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dunidirectional_sequence_rnn.mod.py21 recurrent_weights_data, bias_data, hidden_state_data, output_data, argument
110 bias_data = [ variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantization_utils_test.cc465 float bias_data[] = {4.0, 4.0}; in TEST_F() local
511 float bias_data[] = {4.0}; in TEST_F() local
799 std::vector<float> bias_data = {4.0, 1.0}; in TEST_F() local
857 std::vector<float> bias_data = {4.0, 1.0}; in TEST_F() local

1234