Add 5D date_layout for video input

fastmachinelearning / qonnx

QONNX: Arbitrary-Precision Quantized Neural Networks in ONNX

https://qonnx.readthedocs.io/

Apache License 2.0

124 stars 39 forks source link

Open x7779 opened 3 months ago

x7779 commented 3 months ago

Add a 5-dimension data_layout to handle video input, namely NCDHW or NDHWC.

Modify the src/qonnx/transformation/infer_data_layouts.py to correctly infer the volumetric data.