Add BlockDesc and ProgramDesc to framework.proto

update-doc-pybind
fengjiayi 8 years ago
parent 5b42d2b21b
commit 9a3c6d5f9c

@ -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; }

Loading…
Cancel
Save