未验证 提交 bd58bf3e 编写于 作者: K kavyasrinet 提交者: GitHub

[WIP] Expose Channel in Python and add to VarType (#8465)

Add Channel as a VarType
上级 56d53192
...@@ -112,6 +112,7 @@ message VarType { ...@@ -112,6 +112,7 @@ message VarType {
LOD_TENSOR_ARRAY = 13; LOD_TENSOR_ARRAY = 13;
PLACE_LIST = 14; PLACE_LIST = 14;
READER = 15; READER = 15;
CHANNEL = 16;
} }
required Type type = 1; required Type type = 1;
...@@ -137,6 +138,12 @@ message VarType { ...@@ -137,6 +138,12 @@ message VarType {
message ReaderDesc { repeated LoDTensorDesc lod_tensor = 1; } message ReaderDesc { repeated LoDTensorDesc lod_tensor = 1; }
optional ReaderDesc reader = 5; optional ReaderDesc reader = 5;
message ChannelDesc {
required Type data_type = 1;
required int64 capacity = 2;
}
optional ChannelDesc channel = 6;
} }
message VarDesc { message VarDesc {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册