quantization.hbdk4.get_hbir_output_unflattener

horizon_plugin_pytorch.quantization.hbdk4.get_hbir_output_unflattener(model:Module)

Get an callable func to unflatten model output into origin format.

  • Parameters: model – Hbir model.