search

Samsung / ONE

On-device Neural Engine
Other
414 stars 144 forks source link

[onert] Fix value error when training DepthwiseConv2D #13241

Closed ragmani closed 1 month ago

ragmani commented 1 month ago

This commit fixes value error when training DepthwiseConv2D. DepthwiseConv kernels for training are executed in col major, but LaunchDepthwiseConvBackpropFilterOp's arguments was passed as row major.

ONE-DCO-1.0-Signed-off-by: ragmani ragmani0216@gmail.com

For #13167