Skip to content
Snippets Groups Projects

Update fuseBatchNorm to include ConvDepthWise and MetaOperators

Merged Maxence Naud requested to merge tiling into main
5 files
+ 12
28
Compare changes
  • Side-by-side
  • Inline
Files
5
@@ -51,12 +51,9 @@ public:
return std::make_shared<Erf_Op>(*this);
}
void setBackend(const std::string& name) override {
void setBackend(const std::string& name, DeviceIdx_t device = 0) override {
mImpl = Registrar<Erf_Op>::create(name)(*this);
mOutputs[0]->setBackend(name);
// FIXME: temporary workaround
getInput(0)->setBackend(name);
mOutputs[0]->setBackend(name, device);
}
static const std::vector<std::string> getInputsName(){
Loading