[examples/python/pytorch] added batchnorm cpp extension

2019-07-09 20:59:04 -07:00
parent b7986baffa
commit 63b249c1d6
6 changed files with 214 additions and 206 deletions
--- a/examples/python/tensorflow/shift.cpp
+++ b/examples/python/tensorflow/shift.cpp
@@ -73,7 +73,6 @@ public:
      if(OP == triton::dnn::shift::BPROP)
        std::swap(C, F);
    }
-
  }

  void Compute(OpKernelContext* context){