Skip to content
Snippets Groups Projects

Lenet operators

Merged Houssem ROUIS requested to merge hrouis/aidge_backend_cuda:lenet_operators into dev
5 files
+ 153
151
Compare changes
  • Side-by-side
  • Inline
Files
5
@@ -29,7 +29,7 @@
@@ -29,7 +29,7 @@
namespace Aidge {
namespace Aidge {
class FCImplForward_cuda : public Registrable<FCImplForward_cuda,
class FCImplForward_cuda : public Registrable<FCImplForward_cuda,
std::tuple<DataType>,
std::tuple<DataType>,
void(unsigned int , unsigned int , unsigned int, bool, const void* , const void* , const void* , void*)> {};
void(std::size_t , std::size_t, std::size_t, bool, const void* , const void* , const void* , void*)> {};
class FCImpl_cuda : public OperatorImpl {
class FCImpl_cuda : public OperatorImpl {
private:
private:
// CuDNN specific variables
// CuDNN specific variables
@@ -47,7 +47,7 @@ public:
@@ -47,7 +47,7 @@ public:
// ~FCImpl_cuda();
// ~FCImpl_cuda();
private:
private:
template <class T> void forward_(const Tensor& input0, const Tensor& input1, const Tensor& input2, bool noBias, DimSize_t outChannels);
template <class T> void forward_(const Tensor& input0, const Tensor& input1, const Tensor& input2, bool noBias, std::size_t outChannels);
};
};
namespace {
namespace {
Loading