Fix buffer overrun bug in CPU upsample op (#1501) (#1509)

microsoft · Jul 26, 2019 · 36469ba · 36469ba
1 parent 6df4bc2
commit 36469ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/core/providers/cpu/tensor/upsample.cc b/onnxruntime/core/providers/cpu/tensor/upsample.cc
@@ -342,7 +342,7 @@ Status Upsample<T>::BaseCompute(OpKernelContext* context, const std::vector<floa
   Tensor* Y = context->Output(0, Y_dims);
 
   if (no_scale) {
-    memcpy(Y->MutableDataRaw(), X->DataRaw(), Y->Size() * sizeof(T));
+    memcpy(Y->MutableDataRaw(), X->DataRaw(), Y->Size());
     return Status::OK();
   }