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

[GPU] Added WA for wrong TensorDesc in InferRequest for some outputs (#9734)

Browse Source
This commit is contained in:
Vladimir Paramuzov authored and GitHub committed 2022-01-19 13:06:53 +09:00
1 parent ec3283ebe1
commit 57e918c7b2
1 file changed
+5 -1
Download Patch File Download Diff File Expand all files Collapse all files
src/plugins/intel_gpu/src/plugin/infer_request.cpp
+5 -1
View File
@@ -1042,7 +1042,11 @@ void InferRequest::allocate_outputs() {
for (auto& no : _networkOutputs) {
std::string outputID = m_graph->MapOutputName(no.first);
const cldnn::layout output_layout = m_graph->GetNetwork()->get_output_memory(outputID)->get_layout();
const TensorDesc& desc = no.second->getTensorDesc();
TensorDesc desc = no.second->getTensorDesc();
// Due to some reason TensorDesc in InferRequest contains wrong dims
// while ExecutableNetwork contains proper ones. Thus replace dims with once from exec network
// Can be removed once 76176 is resolved.
desc.setDims(m_graph->GetOutputSize(no.first));
GPU_DEBUG_GET_INSTANCE(debug_config);
GPU_DEBUG_IF(debug_config->verbose >= 2) {