| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_per_channel_quantized_test.cc | 117 const RuntimeShape& input_shape_inference, const std::int8_t* input_data, in PickReasonableMultiplier() argument
124 PickOutputMultiplier(params, input_shape_inference, input_data, in PickReasonableMultiplier()
192 RuntimeShape* input_shape_inference, RuntimeShape* filter_shape_inference, in GenerateValidShapeConfigurations() argument
205 input_shape_inference->BuildFrom( in GenerateValidShapeConfigurations()
212 *input_shape_inference, output_depth, filter_width, filter_height, in GenerateValidShapeConfigurations()
220 *input_shape_inference, *filter_shape_inference, *stride, *stride, in GenerateValidShapeConfigurations()
239 RuntimeShape input_shape_inference; in TryTestOneDepthwiseConv3x3Filter() local
252 &input_shape_inference, &filter_shape_inference, in TryTestOneDepthwiseConv3x3Filter()
259 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestOneDepthwiseConv3x3Filter()
297 input_shape_inference, input_data.data(), filter_shape_inference, in TryTestOneDepthwiseConv3x3Filter()
[all …]
|
| D | depthwiseconv_per_channel_quantized_16x8_test.cc | 116 const RuntimeShape& input_shape_inference, const std::int16_t* input_data, in PickReasonableMultiplier() argument
123 PickOutputMultiplier(params, input_shape_inference, input_data, in PickReasonableMultiplier()
144 RuntimeShape* input_shape_inference, RuntimeShape* filter_shape_inference, in GenerateValidShapeConfigurations() argument
157 input_shape_inference->BuildFrom( in GenerateValidShapeConfigurations()
164 *input_shape_inference, output_depth, filter_width, filter_height, in GenerateValidShapeConfigurations()
194 RuntimeShape input_shape_inference; in TryTestOneDepthwiseConv3x3Filter() local
207 &input_shape_inference, &filter_shape_inference, in TryTestOneDepthwiseConv3x3Filter()
214 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestOneDepthwiseConv3x3Filter()
254 input_shape_inference, input_data.data(), filter_shape_inference, in TryTestOneDepthwiseConv3x3Filter()
263 input_shape_inference, input_data.data(), filter_shape_inference, in TryTestOneDepthwiseConv3x3Filter()
[all …]
|
| D | conv_per_channel_quantized_16x8_test.cc | 117 const RuntimeShape& input_shape_inference, const std::int16_t* input_data, in PickReasonableMultiplier() argument
124 PickOutputMultiplier(params, input_shape_inference, input_data, in PickReasonableMultiplier()
144 int dilation_height_factor, RuntimeShape* input_shape_inference, in GenerateValidShapeConfigurations() argument
157 input_shape_inference->BuildFrom( in GenerateValidShapeConfigurations()
164 *input_shape_inference, output_depth, filter_width, filter_height, in GenerateValidShapeConfigurations()
195 RuntimeShape input_shape_inference; in TryTestOneConvFilter() local
207 dilation_height_factor, &input_shape_inference, in TryTestOneConvFilter()
215 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestOneConvFilter()
260 input_shape_inference, input_data.data(), filter_shape_inference, in TryTestOneConvFilter()
270 input_shape_inference, input_data.data(), filter_shape_inference, in TryTestOneConvFilter()
[all …]
|
| D | depthwiseconv_float_test.cc | 103 RuntimeShape input_shape_inference( in TryTestOneDepthwiseConv() local
109 if (!ComputeConvSizes(input_shape_inference, output_depth, filter_width, in TryTestOneDepthwiseConv()
118 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestOneDepthwiseConv()
141 TestOneDepthwiseConv(op_params, input_shape_inference, input_data.data(), in TryTestOneDepthwiseConv()
|
| D | depthwiseconv_quantized_test.cc | 878 RuntimeShape input_shape_inference( in TryTestDepthwiseConv() local
882 if (!ComputeConvSizes(input_shape_inference, output_depth, filter_width, in TryTestDepthwiseConv()
893 const int input_buffer_size = input_shape_inference.FlatSize(); in TryTestDepthwiseConv()
917 test_param, input_data.data(), input_shape_inference, input_offset, in TryTestDepthwiseConv()
935 test_param, input_data.data(), input_shape_inference, input_offset, in TryTestDepthwiseConv()
|