pytorch/caffe2
Adam Simpkins ccdcfba5de [caffe2] Refactor tensor serialization function (#53404)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53404

This refactors `TensorSerializer::Serialize()` so that we have a separate
helper function for each data type.

This should make it slightly easier in the future to add new serialization
formats for specific data types.
ghstack-source-id: 124085413

Test Plan:
Confirmed the existing tests pass.  This diff is not expected to have any
behavior changes.

Reviewed By: mraway, glamtechie

Differential Revision: D26658204

fbshipit-source-id: 232776262db6486ba845a7ba223e3987053dac27
2021-03-17 12:36:31 -07:00
..
contrib Pass Scalar by reference (#53583) 2021-03-15 23:17:06 -07:00
core [caffe2] Refactor tensor serialization function (#53404) 2021-03-17 12:36:31 -07:00
cuda_rtc
db
distributed
experiments
ideep Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
image Check kernel launches in caffe2/caffe2/image (#52173) 2021-02-12 10:11:41 -08:00
mobile
mpi
observers
onnx
operators [caffe2] add a SerializationOptions field for the save operator (#53402) 2021-03-11 13:02:58 -08:00
opt Avoid of creating a copy of statusString every inference time (#53756) 2021-03-10 16:58:02 -08:00
perfkernels Add install for libCaffe2_perfkernels_avx*.a (#53825) 2021-03-15 08:37:11 -07:00
predictor
proto [caffe2] Support deserializing tensors using alternate serialization formats (#53403) 2021-03-12 11:35:15 -08:00
python [typing] suppress errors in fbcode/caffe2 - batch 2 2021-03-16 16:45:41 -07:00
quantization Clean up future imports for Python 2 (#53349) 2021-03-14 15:56:13 -07:00
queue
serialize Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
sgd [caffe2 and pytorch] replace temp name of new sparse adagrad JIT'ed function in fbgemm (#52193) 2021-02-13 10:23:36 -08:00
share
test
transforms
utils Add guard to run on current thread (#52361) 2021-03-03 11:43:40 -08:00
video Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
.clang-format
__init__.py
c2_aten_srcs.bzl
CMakeLists.txt Move as_view/increment_version to its separate key. (#53342) 2021-03-15 14:47:12 -07:00
README.md
release-notes.md Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
requirements.txt
VERSION_NUMBER

Caffe2

Jenkins Build Status

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

Questions and Feedback

Please use GitHub issues (https://github.com/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai