ONE icon indicating copy to clipboard operation
ONE copied to clipboard

Published 20 hours ago • verified publisher icon Samsung

[onert-micro] how to support on-device training on model with GRU ?

Open chunseoklee opened this issue 7 months ago • 6 comments

GRU operation in circle can be defined in two ways. During conversion from Keras, it may be converted into :

  • Single "Custom" GRU operation as in (onert-micro) https://github.com/Samsung/ONE/pull/11840 + (ONE compiler) https://github.com/Samsung/ONE/issues/12263
  • multi subgraph based on While op as in https://github.com/Samsung/ONE/issues/10465

IMHO, onert-micro is not ready to handle training on multi subgraph.

chunseoklee avatar Jul 09 '24 04:07 chunseoklee