OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT

Browse Source 
Reviewed By: zertosh

Differential Revision: D34412981

fbshipit-source-id: a7aa81c0c69bf731db37813f431d9f6ed6a6a355
This commit is contained in:
CodemodService FBSourceClangFormatLinterBot 2022-02-23 01:35:02 -08:00 committed by Facebook GitHub Bot
parent 736159d415
commit a43ea6d9fc
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/csrc/jit/serialization/export.cpp
View File

@ -21,9 +21,9 @@
#include <atomic>
#include <onnx/checker.h>
#include <onnx/shape_inference/implementation.h>
#include <onnx/onnx_pb.h>
#include <onnx/proto_utils.h>
#include <onnx/shape_inference/implementation.h>
#include <fstream>
#include <memory>
@ -1260,7 +1260,6 @@ void check_onnx_proto(const std::string& proto_string, bool full_check) {
if (full_check) {
onnx::shape_inference::InferShapes(model);
}
}
} // namespace jit

4
torch/csrc/jit/serialization/export.h
View File

@ -61,7 +61,9 @@ export_onnx(
TORCH_API std::string serialize_model_proto_to_string(
const std::shared_ptr<::ONNX_NAMESPACE::ModelProto>& model_proto);
TORCH_API void check_onnx_proto(const std::string& proto_string, bool full_check=false);
TORCH_API void check_onnx_proto(
const std::string& proto_string,
bool full_check = false);
// Serializer for both oldsyle and unified format TorchScript serialization
class TORCH_API ScriptModuleSerializer {