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

[TF][FE] Fixed wrong output naming in AvgPool (#13575)

Browse Source
This commit is contained in:
Georgy Krivoruchko
2022-10-24 16:50:51 +04:00
committed by GitHub
parent 9ea33715a0
commit 94c40d6a82
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -56,7 +56,7 @@ OutputVector translate_avg_pool_op(const NodeContext& node) {
auto_pad);
auto avg_pool_output = avg_pool->output(0);
convert_nchw_to_nhwc(is_nhwc, avg_pool_output, ov::Rank(spatial_dim + 2));
set_node_name(node.get_name(), avg_pool);
set_node_name(node.get_name(), avg_pool_output.get_node_shared_ptr());
return {avg_pool_output};
}