Add qtensors in caffe2 protobuf argument (#18486)
authorzrphercule <zrphercule@gmail.com>
Wed, 27 Mar 2019 18:11:01 +0000 (11:11 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 27 Mar 2019 18:16:40 +0000 (11:16 -0700)
Summary:
We are about to merge onnxifi quantization support soon. Before that, I would like to merge this diff seperately to make sure it doesnt break anything.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18486

Reviewed By: bddppq, houseroad

Differential Revision: D14626419

Pulled By: yinghai

fbshipit-source-id: 504c1eae60be1e629203267b59defb8b69d82c0a

caffe2/proto/caffe2.proto

index ae15521..2b079de 100644 (file)
@@ -163,6 +163,7 @@ message Argument {
   repeated bytes strings = 7;
   repeated TensorProto tensors = 11;
   repeated NetDef nets = 9;
+  repeated QTensorProto qtensors = 12;
 }
 
 // DeviceType that Caffe2 currently supports.