Skip to content
Snippets Groups Projects

feat: Reset to zero for Leaky

Merged Jerome Hue requested to merge jeromeh/aidge_core:leaky-reset into dev
1 unresolved thread
4 files
+ 100
74
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -305,10 +305,17 @@ std::shared_ptr<Node> LSTM(DimSize_t in_channels,
bool noBias = false,
const std::string &name = "");
enum class LeakyReset {
Subtraction,
ToZero
};
std::shared_ptr<MetaOperator_Op> LeakyOp();
std::shared_ptr<Node> Leaky(const int nbTimeSteps,
const float beta,
const float threshold = 1.0,
const LeakyReset resetType = LeakyReset::Subtraction,
const std::string &name = "");
} // namespace Aidge
Loading