Searched refs:bias_shape_inference (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_per_channel_quantized_test.cc | 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()
258 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestOneDepthwiseConv3x3Filter() local
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()
337 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
|
| D | depthwiseconv_per_channel_quantized_16x8_test.cc | 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()
213 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestOneDepthwiseConv3x3Filter() local
255 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
264 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv3x3Filter()
280 filter_shape_inference, filter_data_float.data(), bias_shape_inference, in TryTestOneDepthwiseConv3x3Filter()
|
| D | conv_per_channel_quantized_16x8_test.cc | 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()
214 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestOneConvFilter() local
261 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneConvFilter()
271 filter_data.data(), bias_shape_inference, bias_data.data(), in TryTestOneConvFilter()
289 bias_shape_inference, bias_data_float.data(), in TryTestOneConvFilter()
|
| D | depthwiseconv_float_test.cc | 117 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestOneDepthwiseConv() local
143 bias_shape_inference, bias_data.data(), in TryTestOneDepthwiseConv()
|
| D | depthwiseconv_quantized_test.cc | 892 RuntimeShape bias_shape_inference({1, 1, 1, output_depth}); in TryTestDepthwiseConv() local
919 bias_data.data(), bias_shape_inference, stride, padding_type, pad_width, in TryTestDepthwiseConv()
937 bias_data.data(), bias_shape_inference, stride, padding_type, pad_width, in TryTestDepthwiseConv()
|