mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Make validate private by renaming validate to _validate (#107927)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/107927 Approved by: https://github.com/tugsbayasgalan
This commit is contained in:
parent
f2f82855e2
commit
39854df1d3
|
|
@ -17,7 +17,6 @@ torch.export
|
|||
.. autoclass:: ExportedProgram
|
||||
|
||||
.. automethod:: transform
|
||||
.. automethod:: validate
|
||||
.. automethod:: module
|
||||
|
||||
.. autoclass:: ExportBackwardSignature
|
||||
|
|
|
|||
|
|
@ -143,7 +143,7 @@ class TestExport(TestCase):
|
|||
{},
|
||||
preserve_module_call_signature=("foo.nested", "foo"),
|
||||
)
|
||||
ep.validate()
|
||||
ep._validate()
|
||||
self.assertEqual(len(ep.module_call_graph), 2)
|
||||
# TODO(zhxchen17) unflattener
|
||||
# unflattened = unflatten(export_module)
|
||||
|
|
|
|||
|
|
@ -558,12 +558,7 @@ class ExportedProgram:
|
|||
_assertion_graph = _assertion_graph_res.graph_module
|
||||
_assertion_graph(*args)
|
||||
|
||||
def validate(self):
|
||||
"""
|
||||
.. warning::
|
||||
Do not use.
|
||||
|
||||
"""
|
||||
def _validate(self):
|
||||
# TODO(zhxchen17) check for get_attr
|
||||
# TODO(zhxchen17) check for funcitonal ops
|
||||
for gm in self.graph_module.modules():
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user