Skip to content
Snippets Groups Projects

Fix Relu backward.

Merged Cyril Moineau requested to merge FixReLuBackward into dev
3 files
+ 22
19
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -30,7 +30,7 @@ class ReLUImplForward_cpu
: public Registrable<ReLUImplForward_cpu, std::tuple<DataType, DataType>, void(const std::size_t, const void*, void*)> {
};
class ReLUImplBackward_cpu
: public Registrable<ReLUImplBackward_cpu, std::tuple<DataType, DataType>, void(const std::size_t, const void*, void*)> {
: public Registrable<ReLUImplBackward_cpu, std::tuple<DataType, DataType, DataType>, void(const std::size_t, const void*, const void*, void*)> {
};
class ReLUImpl_cpu : public OperatorImpl {
Loading