IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[TF frontend] use InterpolateMode::LINEAR_ONNX if input rank is 4 (#17381)

Browse Source 
This change mimicks LinearToLinearONNXReplacer transformation in
legacy frontend, where linear interpolate mode is replaced with
linear_onnx due to performance reasons.

Ticket: CVS-108343
This commit is contained in:
Mateusz Tabaka
2023-05-05 17:12:18 +02:00
committed by GitHub
parent 61fc3ae42a
commit 48281376a0
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/frontends/tensorflow_common/src/op/interpolate.cpp
View File

@@ -36,7 +36,12 @@ OutputVector translate_interpolate_op(const NodeContext& node) {
interpolate_attrs.mode = Interpolate::InterpolateMode::NEAREST;
interpolate_attrs.nearest_mode = Interpolate::NearestMode::FLOOR;
} else if (op_type == "ResizeBilinear") {
interpolate_attrs.mode = Interpolate::InterpolateMode::LINEAR;
auto input_rank = images.get_partial_shape().rank();
if (input_rank.is_static() && input_rank.get_length() == 4) {
interpolate_attrs.mode = Interpolate::InterpolateMode::LINEAR_ONNX;
} else {
interpolate_attrs.mode = Interpolate::InterpolateMode::LINEAR;
}
interpolate_attrs.nearest_mode = Interpolate::NearestMode::ROUND_PREFER_FLOOR;
}