diff options
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 { |