Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/micro/kernels/
Dfully_connected_test.cc31 const int* bias_dims_data, const float* bias_data, in TestFullyConnectedFloat()
102 const int32_t* bias_data, const float bias_scale, in TestFullyConnectedQuantized()
193 const float bias_data[] = {1, 2, 3}; in TF_LITE_MICRO_TEST() local
218 const float bias_data[] = {1}; in TF_LITE_MICRO_TEST() local
246 const float bias_data[] = {1, -2, 3}; in TF_LITE_MICRO_TEST() local
304 const int32_t bias_data[] = { in TF_LITE_MICRO_TEST() local
370 const int32_t bias_data[] = { in TF_LITE_MICRO_TEST() local
435 const int32_t bias_data[] = { in TF_LITE_MICRO_TEST() local
500 const int32_t bias_data[] = { in TF_LITE_MICRO_TEST() local
565 const int32_t bias_data[] = { in TF_LITE_MICRO_TEST() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_test.cc99 const Tensor& bias_data, bool transpose_a, in RunMatMulWithBias()
118 const Tensor& lhs_data, const Tensor& rhs_data, const Tensor& bias_data, in RunMatMulWithBiasAndActivation()
221 const Tensor& bias_data, Tensor* out) { in VerifyMatMulWithBias()
228 const Tensor& bias_data, Tensor* out) { in VerifyMatMulWithBias()
251 const Tensor& bias_data, in VerifyConv2DWithBiasAndActivation()
252 Tensor* out) { in VerifyConv2DWithBiasAndActivation()
Dmkl_fused_ops_test.cc206 const Tensor& bias_data, in RunConv2DUnfused()
313 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedConv2D()
314 Tensor* out) { in VerifyFusedConv2D()
918 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBias()
936 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
943 const Tensor& bias_data, Tensor* out) { in VerifyPadAndConv2DWithBiasRelu()
955 const Tensor& bias_data, Tensor* output, in RunMklPadWithFusedConv2DAndBias()
983 const Tensor& bias_data, in RunMklPadWithFusedConv2DAndBiasRelu()
Dmkl_matmul_op_fused.cc142 T* bias_data = const_cast<T*>(bias_tensor.flat<T>().data()); in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h32 const float* bias_data, const RuntimeShape& output_shape, in FullyConnected()
68 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
116 const int32* bias_data, const RuntimeShape& output_shape, in FullyConnected()
169 const int32* bias_data, const RuntimeShape& output_shape, in ShuffledFullyConnected()
Dconv.h31 const float* bias_data, const RuntimeShape& output_shape, in Conv()
104 const int32* bias_data, const RuntimeShape& output_shape, in Conv()
188 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel()
Dlegacy_reference_ops.h43 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
70 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
86 const float* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
102 const float* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
113 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
148 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
168 const int32* bias_data, const Dims<4>& bias_dims, in DepthwiseConv()
191 const int32* bias_data, const Dims<4>& bias_dims, int stride, in DepthwiseConv()
207 const float* bias_data, const Dims<4>& bias_dims, in Conv()
233 const float* bias_data, const Dims<4>& bias_dims, int stride_width, in Conv()
[all …]
Ddepthwiseconv_float.h29 const float* bias_data, const RuntimeShape& output_shape, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h27 const int32* 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* bias_data, const RuntimeShape& output_shape, in DepthwiseConvPerChannel()
126 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
Dconv.h29 const int32* bias_data, const RuntimeShape& output_shape, in ConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/
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_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()
494 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
579 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvWithGivenOutputShift()
716 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvBisectOutputShift()
775 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConv()
856 std::vector<std::int32_t> bias_data(output_depth); in TryTestDepthwiseConv() 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()
265 std::vector<std::int32_t> bias_data(output_depth); in TryTestOneDepthwiseConv3x3Filter() local
/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.h32 const int32* bias_data, const RuntimeShape& output_shape, int8* output_data, in FullyConnected()
Dconv.h35 const int32* bias_data, const RuntimeShape& output_shape, int8* output_data, in ConvPerChannel()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc47 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine() local
118 float* const bias_data = bias_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
Dfuse_binary_into_following_affine.cc75 float* const bias_data = bias_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() 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 …]
Dmultithreaded_conv.h139 const float* bias_data, const RuntimeShape& output_shape, in Conv()
Ddepthwiseconv_multithread.h123 const TS* bias_data, const RuntimeShape& output_shape, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/experimental/ruy/
Dexample.cc48 const float bias_data[] = {1, 0}; in ExampleMulFloatWithBiasAddAndClamp() local
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/
Dtranspose_conv_2d_builder.cc114 std::vector<int> bias_data(output_depth_size); in PopulateSubGraph() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantization_utils_test.cc394 float bias_data[] = {4.0, 4.0}; in TEST_F() local
440 float bias_data[] = {4.0}; in TEST_F() local
689 std::vector<float> bias_data = {4.0, 1.0}; in TEST_F() local
747 std::vector<float> bias_data = {4.0, 1.0}; in TEST_F() local

123