[dnn/shift] now using constant divisions

2019-07-16 21:05:21 -07:00
parent 07c964919c
commit a55b098e88
4 changed files with 17 additions and 13 deletions
--- a/examples/cpp/shift.cpp
+++ b/examples/cpp/shift.cpp
@@ -14,7 +14,7 @@ int main() {

  // initialize default compute device
  auto context = triton::driver::backend::contexts::get_default();
-  auto op = triton::dnn::shift::BPROP;
+  auto op = triton::dnn::shift::FPROP;

  // initialization
  int32_t R = 3, S = 3;