Create call fix.

openvinotoolkit · Jan 30, 2021 · cc64b40 · cc64b40
1 parent 007851c
commit cc64b40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_memory.cpp b/inference-engine/src/mkldnn_plugin/mkldnn_memory.cpp
@@ -152,7 +152,7 @@ void MKLDNNMemory::SetData(memory::data_type dataType, memory::format_tag format
         cpu_memcpy(dataPtr, data, size);
     } else {
         auto memData = this->GetDescriptor().data;
-        std::vector<ptrdiff_t> dims(memData.dims, memData.dims + memData.ndims);
+        memory::dims dims(memData.dims, memData.dims + memData.ndims);
 
         MKLDNNMemory src(this->eng);
         src.Create(dims, dataType, format, data);