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

[TF FE] Correct Deconvolution for NCHW layout (#12751)

Browse Source 
* [TF FE] Correct Deconvolution for NCHW layout

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>

* Revert Deconvolution implementation and work around -1 for SS

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2022-08-25 23:32:55 +03:00 committed by GitHub
parent 7369ce2e4e
commit 3bc7ce1d04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/frontends/tensorflow/src/op/conv_2d_backprop.cpp
View File

@ -77,7 +77,7 @@ OutputVector translate_conv_2d_backprop_input_op(const NodeContext& node) {
// initially think that output shape defined for NCHW layout
auto ss_begin = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{2});
auto ss_end = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{-1});
auto ss_end = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{4});
auto ss_strides = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{1});
// change range of indices for spatial dimensions in case NHWC layout

2
src/frontends/tensorflow/src/op/conv_3d_backprop.cpp
View File

@ -79,7 +79,7 @@ OutputVector translate_conv_3d_backprop_input_v2_op(const NodeContext& node) {
// initially think that output shape defined for NCDHW layout
auto ss_begin = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{2});
auto ss_end = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{-1});
auto ss_end = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{5});
auto ss_strides = make_shared<Constant>(element::i64, Shape{1}, std::vector<int64_t>{1});
// change range of indices for spatial dimensions in case NDHWC layout