remove batch_norm_inference (#2626)

apache · Feb 19, 2019 · 80f8e98 · 80f8e98
1 parent c59a78e
commit 80f8e98
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 135 deletions.
diff --git a/topi/include/topi/nn/batch_norm.h b/topi/include/topi/nn/batch_norm.h
diff --git a/topi/python/topi/nn/__init__.py b/topi/python/topi/nn/__init__.py
@@ -2,7 +2,6 @@
 """Neural network operators"""
 from __future__ import absolute_import as _abs
 
-from .batch_norm import *
 from .conv2d import *
 from .depthwise_conv2d import *
 from .elemwise import *

diff --git a/topi/python/topi/nn/batch_norm.py b/topi/python/topi/nn/batch_norm.py
diff --git a/topi/src/topi.cc b/topi/src/topi.cc
@@ -17,7 +17,6 @@
 #include <topi/reduction.h>
 #include <topi/transform.h>
 
-#include <topi/nn/batch_norm.h>
 #include <topi/nn/bnn.h>
 #include <topi/nn/dense.h>
 #include <topi/nn/dilate.h>
@@ -328,18 +327,6 @@ TVM_REGISTER_GLOBAL("topi.nn.upsampling")
   *rv = nn::upsampling(args[0], args[1], args[2], args[3]);
   });
 
-/* Ops from nn/batch_norm.h */
-TVM_REGISTER_GLOBAL("topi.nn.batch_norm_inference")
-.set_body([](TVMArgs args, TVMRetValue *rv) {
-  *rv = nn::batch_norm_inference(args[0],
-                                 args[1],
-                                 args[2],
-                                 args[3],
-                                 args[4],
-                                 static_cast<double>(args[5]),
-                                 args[6]);
-  });
-
 /* Ops from nn/bnn.h */
 TVM_REGISTER_GLOBAL("topi.nn.binarize_pack")
 .set_body([](TVMArgs args, TVMRetValue *rv) {