Skip to content

[core] Transpose do not change output dformat

  • aidge_core: 0.5.1

Problem description

When forwardDims Transpose.cpp, the output DataFormat is automatically set to DataFormat::Dafault. However, if our input is in NCHW and we apply transposition [0,2,3,1], the output should be in NHWC. I suggest adding a function to DataFormat.hpp

DataFormat getTransposeOutputDataFormat( const DataFormat inputDataFormat, const std::vector<DimSize_t> outputDimsOrder)

And use it to set the correct output DataFormat during the forwardDims() of Transpose.

TODO

  • Method in DataFormat.hpp
  • Change Transpose::forwardDims
  • Add tests
Edited by Wissam Boussella