| 55 | } |
| 56 | |
| 57 | inline TfLitePaddingValues ComputePaddingHeightWidth( |
| 58 | int stride_height, int stride_width, int dilation_rate_height, |
| 59 | int dilation_rate_width, int in_height, int in_width, int filter_height, |
| 60 | int filter_width, TfLitePadding padding, int* out_height, int* out_width) { |
| 61 | *out_width = ComputeOutSize(padding, in_width, filter_width, stride_width, |
| 62 | dilation_rate_width); |
| 63 | *out_height = ComputeOutSize(padding, in_height, filter_height, stride_height, |
| 64 | dilation_rate_height); |
| 65 | |
| 66 | TfLitePaddingValues padding_values; |
| 67 | int offset = 0; |
| 68 | padding_values.height = |
| 69 | ComputePaddingWithOffset(stride_height, dilation_rate_height, in_height, |
| 70 | filter_height, *out_height, &offset); |
| 71 | padding_values.height_offset = offset; |
| 72 | padding_values.width = |
| 73 | ComputePaddingWithOffset(stride_width, dilation_rate_width, in_width, |
| 74 | filter_width, *out_width, &offset); |
| 75 | padding_values.width_offset = offset; |
| 76 | return padding_values; |
| 77 | } |
| 78 | } // namespace tflite |
| 79 | |
| 80 | #endif // TENSORFLOW_LITE_KERNELS_PADDING_H_ |
no test coverage detected