Searched refs:ScriptModuleSerializer (Results 1 – 5 of 5) sorted by relevance
448 void ScriptModuleSerializer::serialize( in serialize()499 void ScriptModuleSerializer::writeArchive( in writeArchive()575 void ScriptModuleSerializer::writeExtraFiles( in writeExtraFiles()604 void ScriptModuleSerializer::updateSourceRangeTags( in updateSourceRangeTags()614 void ScriptModuleSerializer::convertTypes(const at::NamedTypePtr& root_type) { in convertTypes()623 void ScriptModuleSerializer::writeFiles(const std::string& code_dir) { in writeFiles()657 void ScriptModuleSerializer::writeByteCode( in writeByteCode()772 void ScriptModuleSerializer::convertNamedType( in convertNamedType()796 void ScriptModuleSerializer::serialize_unified_format( in serialize_unified_format()826 SerializationStorageContext& ScriptModuleSerializer::storage_context() { in storage_context()[all …]
70 class TORCH_API ScriptModuleSerializer {72 explicit ScriptModuleSerializer( in ScriptModuleSerializer() function85 ~ScriptModuleSerializer() = default;
1020 py::class_<ScriptModuleSerializer>(m, "ScriptModuleSerializer") in initJitScriptBindings()1022 .def("serialize", &ScriptModuleSerializer::serialize_unified_format) in initJitScriptBindings()1025 &ScriptModuleSerializer::writeFiles, in initJitScriptBindings()1029 &ScriptModuleSerializer::storage_context, in initJitScriptBindings()
249 self.script_module_serializer = torch._C.ScriptModuleSerializer(self.zip_file)
934 class ScriptModuleSerializer: