Merge pull request #54 from pfnet-research/okdshin-patch-1

Fix tanh
pfnet-research · Jul 23, 2018 · f503afe · f503afe
2 parents 66f678b + f451ea9
commit f503afe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/menoh/mkldnn/model_core.cpp b/menoh/mkldnn/model_core.cpp
@@ -171,7 +171,7 @@ namespace menoh_impl {
             inplace_primitive_factory_table.insert(
               {"Sqrt", make_sqrt_primitive});
             inplace_primitive_factory_table.insert(
-              {"Tanh", make_sqrt_primitive});
+              {"Tanh", make_tanh_primitive});
 
             std::unordered_map<std::string, primitive_factory>
               primitive_factory_table;