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

Workaround DivideConversion transformation issue in convit model (#15566)

Browse Source 
* Workaround DivideConversion transformation issue in convit model

* Apply suggestions from code review
This commit is contained in:
Maxim Vafin 2023-02-10 14:25:05 +01:00 committed by GitHub
parent d992c6b9c7
commit bfe77d0c65
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/frontends/pytorch/src/op/div.cpp
View File

@ -7,6 +7,7 @@
#include "openvino/op/convert_like.hpp"
#include "openvino/op/divide.hpp"
#include "openvino/op/floor.hpp"
#include "transformations/rt_info/nonconvertible_divide.hpp"
#include "utils.hpp"
using namespace ov::op;
@ -34,6 +35,8 @@ OutputVector translate_div(NodeContext& context) {
}
align_eltwise_input_types(context, x, y, true);
auto res = context.mark_node(std::make_shared<v1::Divide>(x, y, true));
// TODO: ticket 103296; Temporarily disable ConvertDivide transformation
disable_divide_conversion(res);
if (rounding_mode == "floor") {
res = context.mark_node(std::make_shared<v0::Floor>(res));
} else if (rounding_mode == "trunc") {