quantization.hbdk4.get_hbir_input_flattener

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

Get an callable func to flatten model input into a flat tuple of Tensor.

Parameters: model (Module) – Hbir model.

Return type: callable