examples: use correct data type for batch

liyuanpng · Jun 12, 2017 · ebf9bd7 · ebf9bd7
1 parent 095d412
commit ebf9bd7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/simple_net.cpp b/examples/simple_net.cpp
@@ -24,7 +24,7 @@ using namespace mkldnn;
 void simple_net(){
     auto cpu_engine = engine(engine::cpu, 0);
 
-    const uint32_t batch = 8;
+    const int batch = 8;
 
     std::vector<float> net_src(batch * 3 * 227 * 227);
     std::vector<float> net_dst(batch * 96 * 27 * 27);

diff --git a/examples/simple_training_net.cpp b/examples/simple_training_net.cpp
@@ -26,7 +26,7 @@ void simple_net()
 {
     auto cpu_engine = engine(engine::cpu, 0);
 
-    const uint32_t batch = 32;
+    const int batch = 32;
 
     std::vector<float> net_src(batch * 3 * 227 * 227);
     std::vector<float> net_dst(batch * 96 * 27 * 27);