OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity

move the torch/csrc/jit/serialization.h to caffe2 source folder and rename to inline_container.h

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/12781

Reviewed By: dzhulgakov

Differential Revision: D10436151

Pulled By: houseroad

fbshipit-source-id: 7f59eec21df5acbab0ea693e1a1cd4fa152f05e5
This commit is contained in:
Lu Fang 2018-10-18 09:41:30 -07:00 committed by Facebook Github Bot
parent 963b012bd8
commit e240e89984
4 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
torch/csrc/jit/serialization.h → caffe2/serialize/inline_container.h
View File

@ -19,13 +19,13 @@ namespace torch { namespace jit {
// [8 bytes] Magic number - little endian integer that spells 'PYTORCH1' in ASCII
// [8 bytes] Version number - The version of this file format that this file is in.
// this allows us to revise and extend this format
// [56 bytes] Padding/reserved
// [48 bytes] Padding/reserved
//
// After the file header reside N records of the format
// [8 bytes] Tag - this is a tag that identifies the type of this record. The
// values are defined in the RecordTags enum below.
// [8 bytes] size - Size in bytes of the payload of this record
// [56 bytes] Pad/reserved - This space pads out the payload to a 64-byte alignment.
// [48 bytes] Pad/reserved - This space pads out the payload to a 64-byte alignment.
// [size bytes] Payload - The actual raw data for the object serialized in this record
// [size - (size % 64) bytes] Pad/reserved - pad out this record so the next
// one is aligned to 64 bytes
@ -37,7 +37,7 @@ namespace torch { namespace jit {
// as an index into the rest of the file, so
// a reader can use this offset to seek to
// the last record and read the index.
// [56 bytes] Pad/reserved - Pad out the footer s.t. the whole file's size is a
// [48 bytes] Pad/reserved - Pad out the footer s.t. the whole file's size is a
// multiple of 64 bytes.
//
//

5
torch/csrc/jit/export.cpp
View File

@ -1,13 +1,14 @@
#include "torch/csrc/jit/export.h"
#include "torch/csrc/jit/serialization.h"
#include "torch/csrc/autograd/symbolic.h"
#include "onnx/onnx_pb.h"
#include "torch/csrc/onnx/onnx.h"
#include "torch/csrc/utils/functional.h"
#include <torch/csrc/jit/assertions.h>
#include "torch/csrc/jit/passes/dead_code_elimination.h"
#include "caffe2/serialize/inline_container.h"
#include "onnx/onnx_pb.h"
#include <ATen/ATen.h>
#include "c10/util/Optional.h"

5
torch/csrc/jit/import.cpp
View File

@ -1,11 +1,12 @@
#include "torch/csrc/jit/import.h"
#include "torch/csrc/jit/serialization.h"
#include "onnx/onnx_pb.h"
#include "torch/csrc/jit/ir.h"
#include "torch/csrc/utils/functional.h"
#include "torch/csrc/jit/assertions.h"
#include "torch/csrc/jit/operator.h"
#include "caffe2/serialize/inline_container.h"
#include "onnx/onnx_pb.h"
#include <ATen/ATen.h>
#include <unordered_map>

3
torch/csrc/jit/init.cpp
View File

@ -34,10 +34,11 @@
#include "torch/csrc/jit/batched/BatchTensor.h"
#include "torch/csrc/jit/pybind_utils.h"
#include "torch/csrc/jit/function_schema.h"
#include "torch/csrc/jit/serialization.h"
#include "torch/csrc/jit/operator.h"
#include "torch/csrc/jit/fusers/interface.h"
#include "caffe2/serialize/inline_container.h"
#include <pybind11/functional.h>
#include <memory>