Skip to content
Snippets Groups Projects
Commit f5ed784f authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Merged with main

parent e217069c
No related branches found
No related tags found
1 merge request!212Version 0.3.0
Pipeline #55495 passed
...@@ -28,7 +28,7 @@ ...@@ -28,7 +28,7 @@
namespace Aidge { namespace Aidge {
class ILayerNorm_Op : public OperatorTensor, class ILayerNorm_Op : public OperatorTensor,
public Registrable<ILayerNorm_Op, std::string, std::shared_ptr<OperatorImpl>(const ILayerNorm_Op&)> { public Registrable<ILayerNorm_Op, std::string, std::function<std::shared_ptr<OperatorImpl>(const ILayerNorm_Op&)>> {
public: public:
static const std::string Type; static const std::string Type;
...@@ -63,6 +63,7 @@ public: ...@@ -63,6 +63,7 @@ public:
bool forwardDims(bool allowDataDependency = false) override final; bool forwardDims(bool allowDataDependency = false) override final;
void setBackend(const std::string& name, DeviceIdx_t device = 0) override final; void setBackend(const std::string& name, DeviceIdx_t device = 0) override final;
std::set<std::string> getAvailableBackends() const override;
static const std::vector<std::string> getInputsName(){ static const std::vector<std::string> getInputsName(){
return {"data_input", "weight", "bias"}; return {"data_input", "weight", "bias"};
......
...@@ -49,4 +49,8 @@ void Aidge::ILayerNorm_Op::setBackend(const std::string& name, DeviceIdx_t devic ...@@ -49,4 +49,8 @@ void Aidge::ILayerNorm_Op::setBackend(const std::string& name, DeviceIdx_t devic
mOutputs[0]->setBackend(name, device); mOutputs[0]->setBackend(name, device);
getInput(1)->setBackend(name, device); getInput(1)->setBackend(name, device);
getInput(2)->setBackend(name, device); getInput(2)->setBackend(name, device);
} }
\ No newline at end of file
std::set<std::string> Aidge::ILayerNorm_Op::getAvailableBackends() const {
return Registrar<ILayerNorm_Op>::getKeys();
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment