Learning backend cuda
All threads resolved!
All threads resolved!
Compare changes
Files
8- Houssem ROUIS authored
+ 4
− 0
@@ -32,6 +32,10 @@ class And_Op : public OperatorTensor,
Suppport Learning with backend_cuda:
DataProvider.hpp
, DataProvider.cpp
and pybind_DataProvider.cpp
, add backend member to DataProvider;ArgMax.hpp
, ArgMax.cpp
and pybind_ArgMax.cpp
, add ArgMax operator;And.hpp
, And.cpp
and pybind_And.cpp
, add And operator;ReduceSum.hpp
, ReduceSum.cpp
and pybind_ReduceSum.cpp
, add ReduceSum operator;This MR also fixes: #155 (closed)
Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent