diff options
| author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-02-16 04:42:43 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-02-16 04:42:43 +0300 |
| commit | ae600b3ef7cefb82c569444b1477873e4c2229f3 (patch) | |
| tree | 077947625210edef6848eaf117def3f0c53ef069 /src/graph/node_operators_unary.h | |
| parent | b4fdf6134b47220955a41a4b60806c53829eb1c7 (diff) | |
more clean up in backend
Diffstat (limited to 'src/graph/node_operators_unary.h')
| -rw-r--r-- | src/graph/node_operators_unary.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 8390a0c2..a3f27fd2 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -1,10 +1,11 @@ #pragma once -#include "graph/backend_gpu.h" +#include "tensors/tensor.h" +#include "tensors/gpu/backend.h" + #include "graph/node.h" #include "kernels/sparse.h" #include "kernels/tensor_operators.h" -#include "tensors/tensor.h" #include "functional/functional.h" #include "kernels/cudnn_wrappers.h" |