Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
夏有凉风,冬有雪
Paddle
提交
fad48fa6
P
Paddle
项目概览
夏有凉风,冬有雪
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
fad48fa6
编写于
9月 19, 2017
作者:
D
dangqingqing
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add bool type for attr.
上级
6e9337e3
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
29 addition
and
2 deletion
+29
-2
paddle/framework/attribute.cc
paddle/framework/attribute.cc
+18
-0
paddle/framework/attribute.h
paddle/framework/attribute.h
+3
-2
paddle/framework/framework.proto
paddle/framework/framework.proto
+4
-0
python/paddle/v2/framework/op.py
python/paddle/v2/framework/op.py
+4
-0
未找到文件。
paddle/framework/attribute.cc
浏览文件 @
fad48fa6
...
@@ -19,6 +19,10 @@ limitations under the License. */
...
@@ -19,6 +19,10 @@ limitations under the License. */
namespace
paddle
{
namespace
paddle
{
namespace
framework
{
namespace
framework
{
template
<
>
AttrType
AttrTypeID
<
bool
>
()
{
return
BOOL
;
}
template
<
>
template
<
>
AttrType
AttrTypeID
<
int
>
()
{
AttrType
AttrTypeID
<
int
>
()
{
return
INT
;
return
INT
;
...
@@ -32,6 +36,10 @@ AttrType AttrTypeID<std::string>() {
...
@@ -32,6 +36,10 @@ AttrType AttrTypeID<std::string>() {
return
STRING
;
return
STRING
;
}
}
template
<
>
template
<
>
AttrType
AttrTypeID
<
std
::
vector
<
bool
>>
()
{
return
BOOLS
;
}
template
<
>
AttrType
AttrTypeID
<
std
::
vector
<
int
>>
()
{
AttrType
AttrTypeID
<
std
::
vector
<
int
>>
()
{
return
INTS
;
return
INTS
;
}
}
...
@@ -50,6 +58,9 @@ AttrType AttrTypeID<std::vector<std::pair<int, int>>>() {
...
@@ -50,6 +58,9 @@ AttrType AttrTypeID<std::vector<std::pair<int, int>>>() {
Attribute
GetAttrValue
(
const
OpDesc
::
Attr
&
attr_desc
)
{
Attribute
GetAttrValue
(
const
OpDesc
::
Attr
&
attr_desc
)
{
switch
(
attr_desc
.
type
())
{
switch
(
attr_desc
.
type
())
{
case
paddle
::
framework
::
AttrType
::
BOOL
:
{
return
attr_desc
.
b
();
}
case
paddle
::
framework
::
AttrType
::
INT
:
{
case
paddle
::
framework
::
AttrType
::
INT
:
{
return
attr_desc
.
i
();
return
attr_desc
.
i
();
}
}
...
@@ -59,6 +70,13 @@ Attribute GetAttrValue(const OpDesc::Attr& attr_desc) {
...
@@ -59,6 +70,13 @@ Attribute GetAttrValue(const OpDesc::Attr& attr_desc) {
case
paddle
::
framework
::
AttrType
::
STRING
:
{
case
paddle
::
framework
::
AttrType
::
STRING
:
{
return
attr_desc
.
s
();
return
attr_desc
.
s
();
}
}
case
paddle
::
framework
::
AttrType
::
BOOLS
:
{
std
::
vector
<
bool
>
val
(
attr_desc
.
bools_size
());
for
(
int
i
=
0
;
i
<
attr_desc
.
bools_size
();
++
i
)
{
val
[
i
]
=
attr_desc
.
bools
(
i
);
}
return
val
;
}
case
paddle
::
framework
::
AttrType
::
INTS
:
{
case
paddle
::
framework
::
AttrType
::
INTS
:
{
std
::
vector
<
int
>
val
(
attr_desc
.
ints_size
());
std
::
vector
<
int
>
val
(
attr_desc
.
ints_size
());
for
(
int
i
=
0
;
i
<
attr_desc
.
ints_size
();
++
i
)
{
for
(
int
i
=
0
;
i
<
attr_desc
.
ints_size
();
++
i
)
{
...
...
paddle/framework/attribute.h
浏览文件 @
fad48fa6
...
@@ -27,8 +27,9 @@ limitations under the License. */
...
@@ -27,8 +27,9 @@ limitations under the License. */
namespace
paddle
{
namespace
paddle
{
namespace
framework
{
namespace
framework
{
typedef
boost
::
variant
<
boost
::
blank
,
int
,
float
,
std
::
string
,
std
::
vector
<
int
>
,
typedef
boost
::
variant
<
boost
::
blank
,
bool
,
int
,
float
,
std
::
string
,
std
::
vector
<
float
>
,
std
::
vector
<
std
::
string
>
,
std
::
vector
<
bool
>
,
std
::
vector
<
int
>
,
std
::
vector
<
float
>
,
std
::
vector
<
std
::
string
>
,
std
::
vector
<
std
::
pair
<
int
,
int
>>>
std
::
vector
<
std
::
pair
<
int
,
int
>>>
Attribute
;
Attribute
;
...
...
paddle/framework/framework.proto
浏览文件 @
fad48fa6
...
@@ -23,6 +23,8 @@ enum AttrType {
...
@@ -23,6 +23,8 @@ enum AttrType {
FLOATS
=
4
;
FLOATS
=
4
;
STRINGS
=
5
;
STRINGS
=
5
;
INT_PAIRS
=
6
;
INT_PAIRS
=
6
;
BOOL
=
7
;
BOOLS
=
8
;
}
}
message
IntPair
{
message
IntPair
{
...
@@ -44,6 +46,8 @@ message OpDesc {
...
@@ -44,6 +46,8 @@ message OpDesc {
repeated
float
floats
=
7
;
repeated
float
floats
=
7
;
repeated
string
strings
=
8
;
repeated
string
strings
=
8
;
repeated
IntPair
int_pairs
=
9
;
repeated
IntPair
int_pairs
=
9
;
optional
bool
b
=
10
;
repeated
bool
bools
=
6
;
};
};
message
Var
{
message
Var
{
...
...
python/paddle/v2/framework/op.py
浏览文件 @
fad48fa6
...
@@ -89,12 +89,16 @@ class OpDescCreationMethod(object):
...
@@ -89,12 +89,16 @@ class OpDescCreationMethod(object):
new_attr
.
f
=
user_defined_attr
new_attr
.
f
=
user_defined_attr
elif
attr
.
type
==
framework_pb2
.
STRING
:
elif
attr
.
type
==
framework_pb2
.
STRING
:
new_attr
.
s
=
user_defined_attr
new_attr
.
s
=
user_defined_attr
elif
attr
.
type
==
framework_pb2
.
BOOL
:
new_attr
.
b
=
user_defined_attr
elif
attr
.
type
==
framework_pb2
.
INTS
:
elif
attr
.
type
==
framework_pb2
.
INTS
:
new_attr
.
ints
.
extend
(
user_defined_attr
)
new_attr
.
ints
.
extend
(
user_defined_attr
)
elif
attr
.
type
==
framework_pb2
.
FLOATS
:
elif
attr
.
type
==
framework_pb2
.
FLOATS
:
new_attr
.
floats
.
extend
(
user_defined_attr
)
new_attr
.
floats
.
extend
(
user_defined_attr
)
elif
attr
.
type
==
framework_pb2
.
STRINGS
:
elif
attr
.
type
==
framework_pb2
.
STRINGS
:
new_attr
.
strings
.
extend
(
user_defined_attr
)
new_attr
.
strings
.
extend
(
user_defined_attr
)
elif
attr
.
type
==
framework_pb2
.
BOOLS
:
new_attr
.
bools
.
extend
(
user_defined_attr
)
elif
attr
.
type
==
framework_pb2
.
INT_PAIRS
:
elif
attr
.
type
==
framework_pb2
.
INT_PAIRS
:
for
p
in
user_defined_attr
:
for
p
in
user_defined_attr
:
pair
=
new_attr
.
int_pairs
.
add
()
pair
=
new_attr
.
int_pairs
.
add
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录