Samsung / ONE

On-device Neural Engine
Other
429 stars 157 forks source link

[onert] Enable `ConstantInsertionPass` for training #13350

Closed ragmani closed 2 months ago

ragmani commented 3 months ago

What

Let's enable ConstantInsertionPass again for training of models that share a constant data in multiple operation nodes.

Why

There are some models that share a constant operand like the Conv2D nodes (@0, @1), (@2, @3), (@4, @5), (@6, @7) in the below log.

PR

ragmani commented 3 months ago

This issue may related to #13223, because the above model worked well in the past.

ragmani commented 2 months ago

After #13514

But #13514 has memory issues in inference, it's going to be applied to training only.

ragmani commented 2 months ago

13514 is replaced with #13517.

ragmani commented 2 months ago

Done.