From d578b25db8ab9afe7720ca3a2863de91bf069a5a Mon Sep 17 00:00:00 2001
From: qiaolongfei <qiaolongfei@baidu.com>
Date: Wed, 6 Sep 2017 12:09:17 -0700
Subject: [PATCH] add var desc proto

---
 paddle/framework/framework.proto | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/paddle/framework/framework.proto b/paddle/framework/framework.proto
index 368136a9729..dfcb5fb6210 100644
--- a/paddle/framework/framework.proto
+++ b/paddle/framework/framework.proto
@@ -87,3 +87,24 @@ message OpProto {
   repeated Attr attrs = 4;
   required string comment = 5;
 }
+
+enum DataType {
+  BOOL = 0;
+  INT16 = 1;
+  INT32 = 2;
+  INT64 = 3;
+  FP16 = 4;
+  FP32 = 5;
+  FP64 = 6;
+}
+
+message LoDTensorDesc {
+  required DataType data_type = 1;
+  repeated int32 dims = 2; // [UNK, 640, 480] is saved as [-1, 640, 480]
+  optional int32 lod_level = 3 [ default = 0 ];
+}
+
+message VarDesc {
+  required string name = 1;
+  optional LoDTensorDesc lod_tensor = 2;
+}
-- 
GitLab