Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
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()
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()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
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
Dfully_connected_tester.cc143 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
195 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
Dconv_2d_tester.cc160 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
218 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
Ddepthwise_conv_2d_tester.cc146 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
199 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
/external/tensorflow/tensorflow/lite/kernels/internal/
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
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
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
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
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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
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()
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()
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 …]
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()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
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()
Dfully_connected.h36 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnectedPerChannel()
105 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
/external/tensorflow/tensorflow/core/kernels/
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()
/external/ruy/ruy/
Dmul_params_test.cc42 std::int32_t bias_data[1]; in TEST() local
Dtest_overflow_dst_zero_point.cc66 std::int32_t bias_data[1] = {bias_value}; in TestOverflowingAdditionOfDestinationZeroPoint() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h29 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/core/kernels/mkl/
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 …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc60 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine() local
132 float* const bias_data = bias_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
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 …]

123