onnx_diagnostic.helpers.rt_helper

onnx_diagnostic.helpers.rt_helper.make_feeds(proto: ModelProto | List[str], inputs: Any, use_numpy: bool = False, copy: bool = False, check_flatten: bool = True) → Dict[str, ndarray | Tensor]

Serializes the inputs to produce feeds expected by onnxruntime.InferenceSession.

Parameters:

• proto – onnx model or list of names

• inputs – any kind of inputs

• use_numpy – if True, converts torch tensors into numpy arrays

• copy – a copy is made, this should be the case if the inputs is ingested by OrtValue

• check_flatten – if True, checks the torch.utils._pytree.tree_flatten returns the same number of outputs

Returns:

feeds dictionary