tensorplay.onnx.verify_model

tensorplay.onnx.verify_model(model: Any, *, expected: Any, input_names: Sequence[str], example_inputs: dict[str, Any], rtol: float = 0.0001, atol: float = 1e-05, raise_on_mismatch: bool = True) → VerificationResult[source]

Compare model under onnxruntime against the eager result.

Parameters:

• model – the built ModelProto.

• expected – what eager TensorPlay produced for example_inputs — a tensor, or an arbitrarily nested tuple/list of them.

• input_names – ONNX graph input names, in placeholder order.

• example_inputs – placeholder name -> example tensor.

• rtol/atol – tolerances forwarded to numpy.allclose().

• raise_on_mismatch – raise VerificationError instead of returning a failed result.