|
|
|
@ -23,6 +23,7 @@ enum AttrType {
|
|
|
|
|
FLOATS = 4;
|
|
|
|
|
STRINGS = 5;
|
|
|
|
|
INT_PAIRS = 6;
|
|
|
|
|
BLOCK = 7;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
message IntPair {
|
|
|
|
@ -44,6 +45,7 @@ message OpDesc {
|
|
|
|
|
repeated float floats = 7;
|
|
|
|
|
repeated string strings = 8;
|
|
|
|
|
repeated IntPair int_pairs = 9;
|
|
|
|
|
optional int32 block = 10;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
message Var {
|
|
|
|
@ -108,3 +110,12 @@ message VarDesc {
|
|
|
|
|
required string name = 1;
|
|
|
|
|
optional LoDTensorDesc lod_tensor = 2;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
message BlockDesc {
|
|
|
|
|
required int32 idx = 1;
|
|
|
|
|
required int32 parent_idx = 2;
|
|
|
|
|
repeated VarDesc vars = 3;
|
|
|
|
|
repeated OpDesc ops = 4;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
message ProgramDesc { repeated BlockDesc blocks = 1; }
|
|
|
|
|