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