[dnn] Adding batchnorm

2019-07-08 18:44:37 -07:00
parent b0cf3143c5
commit f9db0449b7
42 changed files with 682 additions and 1763 deletions
--- a/examples/python/pytorch/conv.cpp
+++ b/examples/python/pytorch/conv.cpp
@@ -56,7 +56,7 @@ torch::Tensor conv_common(
                                                stride_d, stride_h, stride_w,
                                                pad_d, pad_h, pad_w,
                                                1, 1, 1,
-                                                ty, has_bias)).first->second.get();
+                                                "fp32", "fp32", ty, has_bias)).first->second.get();
  else
    configuration = m_config.at(key).get();