Fixed bug in from_float_vector (#4180)

3665d85f · Geoffrey Wenger · Scott Cyphers · c748bda4 · 3665d85f
Commit 3665d85f authored Jan 14, 2020 by Geoffrey Wenger Committed by Scott Cyphers Jan 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

bfloat16.cpp src/ngraph/type/bfloat16.cpp +2 -1

No files found.
--- a/src/ngraph/type/bfloat16.cpp
+++ b/src/ngraph/type/bfloat16.cpp
@@ -55,7 +55,8 @@ std::vector<float> bfloat16::to_float_vector(const std::vector<bfloat16>& v_bf16

 std::vector<bfloat16> bfloat16::from_float_vector(const std::vector<float>& v_f32)
 {
-    std::vector<bfloat16> v_bf16(v_f32.size());
+    std::vector<bfloat16> v_bf16;
+    v_bf16.reserve(v_f32.size());
    for (float a : v_f32)
    {
        v_bf16.push_back(static_cast<bfloat16>(a));