This commit is contained in:
Philippe Tillet
2019-07-16 12:59:27 -07:00
parent f50d7a420a
commit 7d1797cd32
6 changed files with 27 additions and 28 deletions

View File

@@ -14,7 +14,7 @@ int main() {
// initialize default compute device
auto context = triton::driver::backend::contexts::get_default();
auto op = triton::dnn::shift::WGRAD;
auto op = triton::dnn::shift::FPROP;
// initialization
int32_t R = 3, S = 3;