Searched refs:export_writer (Results 1 – 2 of 2) sorted by relevance
73 caffe2::serialize::PyTorchStreamWriter& export_writer) in ScriptModuleSerializer() argument74 : writer_(export_writer) {} in ScriptModuleSerializer()
935 def __init__(self, export_writer: PyTorchFileWriter) -> None: ...