Home
last modified time | relevance | path

Searched refs:output_shape_inference (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_per_channel_quantized_16x8_test.cc119 const RuntimeShape& output_shape_inference, in PickReasonableMultiplier() argument
125 bias_shape_inference, bias_data, output_shape_inference, in PickReasonableMultiplier()
145 RuntimeShape* output_shape_inference, int* pad_width, int* pad_height, in GenerateValidShapeConfigurations() argument
166 output_shape_inference, pad_width, pad_height)); in GenerateValidShapeConfigurations()
196 RuntimeShape output_shape_inference; in TryTestOneDepthwiseConv3x3Filter() local
208 &output_shape_inference, &pad_width, &pad_height, &stride); in TryTestOneDepthwiseConv3x3Filter()
211 const int output_depth = output_shape_inference.Dims(3); in TryTestOneDepthwiseConv3x3Filter()
216 const int output_buffer_size = output_shape_inference.FlatSize(); in TryTestOneDepthwiseConv3x3Filter()
256 output_shape_inference, output_multiplier.data(), output_shift.data(), in TryTestOneDepthwiseConv3x3Filter()
265 output_shape_inference, reference_output_data.data()); in TryTestOneDepthwiseConv3x3Filter()
[all …]
Ddepthwiseconv_per_channel_quantized_test.cc120 const RuntimeShape& output_shape_inference, in PickReasonableMultiplier() argument
126 bias_shape_inference, bias_data, output_shape_inference, in PickReasonableMultiplier()
193 RuntimeShape* output_shape_inference, int* pad_width, int* pad_height, in GenerateValidShapeConfigurations() argument
214 output_shape_inference, pad_width, pad_height)); in GenerateValidShapeConfigurations()
222 depth_multiplier, *output_shape_inference, 0); in GenerateValidShapeConfigurations()
241 RuntimeShape output_shape_inference; in TryTestOneDepthwiseConv3x3Filter() local
253 &output_shape_inference, &pad_width, &pad_height, &stride); in TryTestOneDepthwiseConv3x3Filter()
256 const int output_depth = output_shape_inference.Dims(3); in TryTestOneDepthwiseConv3x3Filter()
261 const int output_buffer_size = output_shape_inference.FlatSize(); in TryTestOneDepthwiseConv3x3Filter()
299 output_shape_inference, output_multiplier.data(), output_shift.data(), in TryTestOneDepthwiseConv3x3Filter()
[all …]
Dconv_per_channel_quantized_16x8_test.cc120 const RuntimeShape& output_shape_inference, in PickReasonableMultiplier() argument
126 bias_shape_inference, bias_data, output_shape_inference, in PickReasonableMultiplier()
145 RuntimeShape* filter_shape_inference, RuntimeShape* output_shape_inference, in GenerateValidShapeConfigurations() argument
166 output_shape_inference, pad_width, pad_height)); in GenerateValidShapeConfigurations()
197 RuntimeShape output_shape_inference; in TryTestOneConvFilter() local
208 &filter_shape_inference, &output_shape_inference, &pad_width, in TryTestOneConvFilter()
212 const int output_depth = output_shape_inference.Dims(3); in TryTestOneConvFilter()
217 const int output_buffer_size = output_shape_inference.FlatSize(); in TryTestOneConvFilter()
262 output_shape_inference, output_multiplier.data(), output_shift.data(), in TryTestOneConvFilter()
272 output_shape_inference, reference_output_data.data()); in TryTestOneConvFilter()
[all …]
Ddepthwiseconv_float_test.cc105 RuntimeShape output_shape_inference; in TryTestOneDepthwiseConv() local
112 &output_shape_inference, &pad_width, &pad_height)) { in TryTestOneDepthwiseConv()
144 output_shape_inference); in TryTestOneDepthwiseConv()
Ddepthwiseconv_quantized_test.cc880 RuntimeShape output_shape_inference; in TryTestDepthwiseConv() local
885 &output_shape_inference, &pad_width, &pad_height)) { in TryTestDepthwiseConv()
888 TFLITE_DCHECK_EQ(output_depth, output_shape_inference.Dims(3)); in TryTestDepthwiseConv()
922 output_activation_min, output_activation_max, output_shape_inference); in TryTestDepthwiseConv()
940 output_activation_min, output_activation_max, output_shape_inference); in TryTestDepthwiseConv()