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