Skip to content
Snippets Groups Projects

Fixed issues for LSTM

Merged Olivier BICHLER requested to merge lstm into dev
5 files
+ 43
26
Compare changes
  • Side-by-side
  • Inline
Files
5
@@ -50,7 +50,7 @@ class StackOp
@@ -50,7 +50,7 @@ class StackOp
public:
public:
static const std::string s_type;
static const std::string s_type;
StackOp(std::uint32_t maxElements);
StackOp(std::uint32_t maxElements = 0);
/**
/**
* @brief Copy-constructor. Copy the operator attributes and its output
* @brief Copy-constructor. Copy the operator attributes and its output
@@ -71,6 +71,7 @@ class StackOp
@@ -71,6 +71,7 @@ class StackOp
std::set<std::string> getAvailableBackends() const override;
std::set<std::string> getAvailableBackends() const override;
 
bool dimsForwarded() const override final;
bool forwardDims(bool allowDataDependency = false) override final;
bool forwardDims(bool allowDataDependency = false) override final;
void forward() override;
void forward() override;
@@ -87,14 +88,14 @@ class StackOp
@@ -87,14 +88,14 @@ class StackOp
}
}
static const std::vector<std::string> getInputsName() {
static const std::vector<std::string> getInputsName() {
return {"data_input"};
return {"data_input", "max_elements"};
}
}
static const std::vector<std::string> getOutputsName() {
static const std::vector<std::string> getOutputsName() {
return {"data_output"};
return {"data_output"};
}
}
};
};
std::shared_ptr<Node> stack(std::uint32_t maxElements,
std::shared_ptr<Node> Stack(std::uint32_t maxElements = 0,
const std::string &name = "");
const std::string &name = "");
} // namespace Aidge
} // namespace Aidge
Loading