File tree Expand file tree Collapse file tree 3 files changed +93
-0
lines changed Expand file tree Collapse file tree 3 files changed +93
-0
lines changed Original file line number Diff line number Diff line change @@ -27,6 +27,7 @@ add_subdirectory(fleet)
2727add_subdirectory (io)
2828#ddim lib
2929proto_library(framework_proto SRCS framework .proto)
30+ proto_library(op_def_proto SRCS op_def.proto)
3031proto_library(heter_service_proto SRCS heter_service.proto)
3132proto_library(data_feed_proto SRCS data_feed.proto)
3233proto_library(trainer_desc_proto SRCS trainer_desc.proto DEPS framework_proto
Original file line number Diff line number Diff line change 1+ /* Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
2+
3+ Licensed under the Apache License, Version 2.0 (the "License");
4+ you may not use this file except in compliance with the License.
5+ You may obtain a copy of the License at
6+
7+ http://www.apache.org/licenses/LICENSE-2.0
8+
9+ Unless required by applicable law or agreed to in writing, software
10+ distributed under the License is distributed on an "AS IS" BASIS,
11+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ See the License for the specific language governing permissions and
13+ limitations under the License. */
14+
15+ syntax = "proto2" ;
16+
17+ import "framework.proto" ;
18+ package paddle.framework.proto ;
19+
20+ message OpDef {
21+
22+ message VarDef {
23+ required string name = 1 ;
24+
25+ // For the type of input / output variables.
26+ reserved 2 ;
27+ }
28+
29+ message AttrDef {
30+ required string name = 1 ;
31+ required AttrType type = 2 ;
32+ }
33+
34+ message Desc {
35+ repeated VarDef inputs = 1 ;
36+ repeated VarDef outputs = 2 ;
37+ repeated AttrDef attrs = 3 ;
38+ }
39+
40+ required string type = 1 ;
41+ required Desc def = 2 ;
42+ optional Desc extra = 3 ;
43+ }
Original file line number Diff line number Diff line change 1+ type: " while"
2+ def {
3+ inputs {
4+ name: " X"
5+ }
6+ inputs {
7+ name: " Condition"
8+ }
9+ outputs {
10+ name: " Out"
11+ }
12+ outputs {
13+ name: " StepScopes"
14+ }
15+ attrs {
16+ name: " sub_block"
17+ type: BLOCK
18+ }
19+ }
20+ extra {
21+ attrs {
22+ name: " is_test"
23+ type: BOOLEAN
24+ }
25+ attrs {
26+ name: " skip_eager_deletion_vars"
27+ type: STRINGS
28+ }
29+ attrs {
30+ name: " op_role"
31+ type: INT
32+ }
33+ attrs {
34+ name: " op_role_var"
35+ type: STRINGS
36+ }
37+ attrs {
38+ name: " op_namescope"
39+ type: STRING
40+ }
41+ attrs {
42+ name: " op_callstack"
43+ type: STRINGS
44+ }
45+ attrs {
46+ name: " op_device"
47+ type: STRING
48+ }
49+ }
You can’t perform that action at this time.
0 commit comments