Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h34 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
80 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl()
152 const float* bias_data; member
168 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h30 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
66 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
114 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
168 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()
Dconv.h28 const float* bias_data, const RuntimeShape& output_shape, in Conv()
104 const int32_t* bias_data, const RuntimeShape& output_shape, in Conv()
191 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel()
Dlegacy_reference_ops.h44 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
71 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
87 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
103 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
114 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
149 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
169 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
192 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
208 const float* bias_data, const Dims<4>& bias_dims, in Conv()
234 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
[all …]
/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
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
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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h27 const int32_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
69 const int64_t* bias_data, const RuntimeShape& output_shape, in FullyConnected()
Ddepthwise_conv.h27 const int32_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
127 const std::int64_t* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
212 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
Dconv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in ConvPerChannel()
134 const std::int64_t* bias_data, const RuntimeShape& output_shape, in ConvPerChannel()
Dtranspose_conv.h29 const int32_t* bias_data, const RuntimeShape& output_shape, in TransposeConv()
128 const std::int64_t* bias_data, const RuntimeShape& output_shape, in TransposeConv()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h45 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConvHybridGeneral()
263 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding()
322 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl()
382 const TS* bias_data; member
396 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
Dfully_connected.h34 const int32* bias_data, const RuntimeShape& output_shape, int8* output_data, in FullyConnected()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dfully_connected_tester.cc133 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
180 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
Dconv_2d_tester.cc148 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
206 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
Ddepthwise_conv_2d_tester.cc134 std::vector<uint16_t> bias_data(OutputChannels()); in CreateTfLiteModel() local
187 std::vector<float> bias_data(OutputChannels()); in CreateTfLiteModel() local
/external/ruy/ruy/
Dmul_params_test.cc42 std::int32_t bias_data[1]; in TEST() local
/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()
255 const Tensor& bias_data, in VerifyConv2DWithBiasAndActivation()
256 Tensor* out) { in VerifyConv2DWithBiasAndActivation()
/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.cc216 const Tensor& bias_data, in RunConv2DUnfused()
359 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedConv2D()
360 Tensor* out, const int padding) { in VerifyFusedConv2D()
556 const Tensor& bias_data, in RunDepthwiseConv2DUnfused()
682 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedDepthwiseConv2D()
683 Tensor* out, const int padding) { in VerifyFusedDepthwiseConv2D()
1481 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBias()
1499 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
1506 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
1518 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.h140 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
172 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
188 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
204 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
217 const int32* bias_data, const RuntimeShape& output_shape, in LegacyDepthwiseConvWithRounding()
271 const int32* bias_data, const RuntimeShape& output_shape, in LegacyDepthwiseConvImpl()
283 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
323 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
343 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
366 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
[all …]

123