diff options
| author | Lu Fang <lufang@fb.com> | 2018-11-19 14:29:31 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-11-19 14:34:05 -0800 |
| commit | f34c848f525ca346637cb57e8b7a607285769e50 (patch) | |
| tree | a9ae97b60aa9cbde2722a48621e6b33b2e67b132 /caffe2/proto | |
| parent | 7fd1ea6ab792ac572f6e1261cc995d6447e66705 (diff) | |
| download | pytorch-f34c848f525ca346637cb57e8b7a607285769e50.tar.gz pytorch-f34c848f525ca346637cb57e8b7a607285769e50.tar.bz2 pytorch-f34c848f525ca346637cb57e8b7a607285769e50.zip | |
Store the optimize flag in module (#14166)
Summary:
When the save/load of script module, we store optimize flag in module instead of encoding it in method.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14166
Reviewed By: ezyang
Differential Revision: D13117577
Pulled By: dzhulgakov
fbshipit-source-id: dc322948bda0ac5809d8ef9a345497ebb8f33a61
Diffstat (limited to 'caffe2/proto')
| -rw-r--r-- | caffe2/proto/torch.proto | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/caffe2/proto/torch.proto b/caffe2/proto/torch.proto index 3e70cc61e9..f7e2e8cace 100644 --- a/caffe2/proto/torch.proto +++ b/caffe2/proto/torch.proto @@ -56,6 +56,10 @@ message ModuleDef { // from the main method. optional string name = 6; + + // whether apply the optimizations to this module, only applicable to + // script modules + optional bool optimize = 7; } enum ProtoVersion { |