Search Docs
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