update benchmark and test code

pfnet-research · Sep 27, 2018 · 9dac97a · 9dac97a
1 parent fc3ec79
commit 9dac97a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/benchmark/vgg16_benchmark.cpp b/benchmark/vgg16_benchmark.cpp
@@ -35,7 +35,7 @@ int main(int argc, char** argv) {
     menoh::variable_profile_table_builder vpt_builder;
     vpt_builder.add_input_profile(conv1_1_in_name, menoh::dtype_t::float_,
                                   {batch_size, 3, 224, 224});
-    vpt_builder.add_output_profile(softmax_out_name, menoh::dtype_t::float_);
+    vpt_builder.add_output_name(softmax_out_name);
     auto vpt = vpt_builder.build_variable_profile_table(model_data);
 
     // Build model

diff --git a/test/backend.hpp b/test/backend.hpp
@@ -61,9 +61,8 @@ namespace menoh {
             std::vector<float> data;
             std::tie(std::ignore, output_dims, data) =
               menoh_impl::load_np_array(output_filename);
-            dtype_t dtype = dtype_t::float_; // TODO other dtype
             true_output_table.insert({output_name, data});
-            vpt_builder.add_output_profile(output_name, dtype);
+            vpt_builder.add_output_name(output_name);
         }
 
         auto vpt = vpt_builder.build_variable_profile_table(model_data);

diff --git a/test/mkldnn_with_generic_fallback_backend.cpp b/test/mkldnn_with_generic_fallback_backend.cpp
@@ -69,15 +69,14 @@ namespace menoh {
             std::string output_name;
             std::tie(output_name, std::ignore) = gemm_true_output_filename;
             model_data.add_output_name_to_current_node(output_name);
-            dtype_t dtype = dtype_t::float_; // TODO other dtype
-            vpt_builder.add_output_profile(output_name, dtype);
+            vpt_builder.add_output_name(output_name);
         }
 
         {
             model_data.add_new_node("Relu");
             model_data.add_input_name_to_current_node("gemm_out");
             model_data.add_output_name_to_current_node("relu_out");
-            vpt_builder.add_output_profile("relu_out", dtype_t::float_);
+            vpt_builder.add_output_name("relu_out");
         }
 
         auto vpt = vpt_builder.build_variable_profile_table(model_data);