You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
343 B
13 lines
343 B
"""A set of tools to verify the correctness of ONNX models."""
|
|
|
|
__all__ = ["VerificationInfo", "verify_onnx_program"]
|
|
|
|
from torch.onnx._internal.exporter._verification import (
|
|
VerificationInfo,
|
|
verify_onnx_program,
|
|
)
|
|
|
|
|
|
VerificationInfo.__module__ = "torch.onnx.verification"
|
|
verify_onnx_program.__module__ = "torch.onnx.verification"
|