[rebase] Rebase fix

This patch adds rebase fix.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/nntrainer/graph/network_graph.cpp b/nntrainer/graph/network_graph.cpp
index 6bb0135..c8cbb41 100644 (file)
--- a/nntrainer/graph/network_graph.cpp
+++ b/nntrainer/graph/network_graph.cpp
@@ -426,6 +426,9 @@ void NetworkGraph::setBatchSize(unsigned int batch_size) {
     return;
 
   this->batch_size = batch_size;
+  if (!input_list.empty() and input_list[0]->getDim().batch() == batch_size)
+    return;
+
   auto allocated = tensor_manager->isAllocated();
 
   if (allocated)