Set correct output shape

fbcc4792 · Paul · 5f7826da · fbcc4792
Commit fbcc4792 authored May 24, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/tf/tf.cpp src/tf/tf.cpp +1 -0

No files found.
--- a/src/tf/tf.cpp
+++ b/src/tf/tf.cpp
@@ -1010,6 +1010,7 @@ program parse_tf(const std::string& name, bool is_nhwc)
 #else
    parser.parse_from(input);
 #endif
+    parser.to_nchw(std::prev(parser.prog.end()));
    return std::move(parser.prog);
 }