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 (Module) – Hbir model.

Return type: callable