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

[GPU] Fix gather data type issue (#12085) (#12085)

Browse Source
This commit is contained in:
Paul Youngsoo Ahn 2022-07-08 21:18:28 +09:00 committed by GitHub
parent 73381e976e
commit 7fe3ab0328
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/intel_gpu/src/graph/gather.cpp
View File

@ -53,6 +53,10 @@ layout gather_inst::calc_output_layout(gather_node const& node) {
}
}
auto output_type = input_layout.data_type;
if (node.has_fused_primitives()) {
output_type = node.get_fused_output_layout().data_type;
}
return layout{output_type,
output_format,
tensor(format::get_default_format(dims_converted.size()), dims_converted)};