Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2301_77200941
mindspore
提交
2224fa09
M
mindspore
项目概览
2301_77200941
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
2224fa09
编写于
5月 21, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
5月 21, 2020
浏览文件
操作
浏览文件
下载
差异文件
!1302 Complete vm ops for BinaryCrossEntropy and BinaryCrossEntropyGrad
Merge pull request !1302 from lihongkang/master
上级
b96a2ed8
cf543382
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
100 addition
and
0 deletion
+100
-0
mindspore/ops/_op_impl/tbe/__init__.py
mindspore/ops/_op_impl/tbe/__init__.py
+2
-0
mindspore/ops/_op_impl/tbe/binary_cross_entropy.py
mindspore/ops/_op_impl/tbe/binary_cross_entropy.py
+41
-0
mindspore/ops/_op_impl/tbe/binary_cross_entropy_grad.py
mindspore/ops/_op_impl/tbe/binary_cross_entropy_grad.py
+44
-0
tests/ut/python/ops/test_ops.py
tests/ut/python/ops/test_ops.py
+13
-0
未找到文件。
mindspore/ops/_op_impl/tbe/__init__.py
浏览文件 @
2224fa09
...
...
@@ -189,3 +189,5 @@ from .pack import _pack_tbe
from
.unpack
import
_unpack_tbe
from
.prelu
import
_prelu_tbe
from
.prelu_grad
import
_prelu_grad_tbe
from
.binary_cross_entropy
import
_binary_cross_entropy_tbe
from
.binary_cross_entropy_grad
import
_binary_cross_entropy_grad_tbe
mindspore/ops/_op_impl/tbe/binary_cross_entropy.py
0 → 100644
浏览文件 @
2224fa09
# Copyright 2020 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
"""BinaryCrossEntropy op"""
from
mindspore.ops.op_info_register
import
op_info_register
,
TBERegOp
,
DataType
binary_cross_entropy_op_info
=
TBERegOp
(
"BinaryCrossEntropy"
)
\
.
fusion_type
(
"OPAQUE"
)
\
.
async_flag
(
False
)
\
.
binfile_name
(
"binary_cross_entropy.so"
)
\
.
compute_cost
(
10
)
\
.
kernel_name
(
"binary_cross_entropy"
)
\
.
partial_flag
(
True
)
\
.
attr
(
"reduction"
,
"optional"
,
"str"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"y"
,
False
,
"required"
,
"all"
)
\
.
input
(
2
,
"weight"
,
False
,
"optional"
,
"all"
)
\
.
output
(
0
,
"output"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
get_op_info
()
@
op_info_register
(
binary_cross_entropy_op_info
)
def
_binary_cross_entropy_tbe
():
"""BinaryCrossEntropy TBE register"""
return
mindspore/ops/_op_impl/tbe/binary_cross_entropy_grad.py
0 → 100644
浏览文件 @
2224fa09
# Copyright 2020 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
"""BinaryCrossEntropyGrad op"""
from
mindspore.ops.op_info_register
import
op_info_register
,
TBERegOp
,
DataType
binary_cross_entropy_grad_op_info
=
TBERegOp
(
"BinaryCrossEntropyGrad"
)
\
.
fusion_type
(
"OPAQUE"
)
\
.
async_flag
(
False
)
\
.
binfile_name
(
"binary_cross_entropy_grad.so"
)
\
.
compute_cost
(
10
)
\
.
kernel_name
(
"binary_cross_entropy_grad"
)
\
.
partial_flag
(
True
)
\
.
attr
(
"reduction"
,
"optional"
,
"str"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"y"
,
False
,
"required"
,
"all"
)
\
.
input
(
2
,
"grad_output"
,
False
,
"required"
,
"all"
)
\
.
input
(
3
,
"weight"
,
False
,
"optional"
,
"all"
)
\
.
output
(
0
,
"output"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
get_op_info
()
@
op_info_register
(
binary_cross_entropy_grad_op_info
)
def
_binary_cross_entropy_grad_tbe
():
"""BinaryCrossEntropyGrad TBE register"""
return
tests/ut/python/ops/test_ops.py
浏览文件 @
2224fa09
...
...
@@ -972,6 +972,19 @@ test_case_nn_ops = [
'desc_inputs'
:
[[
3
,
3
],
[
3
,
3
],
Tensor
(
0.001
,
mstype
.
float32
),
[
3
,
3
],
Tensor
(
0.1
,
mstype
.
float32
),
[
3
,
3
]],
'desc_bprop'
:
[
3
,
3
],
'skip'
:
[
'backward'
]}),
(
'BinaryCrossEntropy'
,
{
'block'
:
P
.
BinaryCrossEntropy
(),
'desc_inputs'
:
[
Tensor
([[
0.3
,
0.8
],
[
0.4
,
0.3
]],
mstype
.
float16
),
Tensor
([[
0.4
,
1.2
],
[
-
0.4
,
-
0.9
]],
mstype
.
float16
),
Tensor
([[
-
1.4
,
-
0.7
],
[
0.9
,
0.7
]],
mstype
.
float16
)],
'desc_bprop'
:
[]}),
(
'BinaryCrossEntropyGrad'
,
{
'block'
:
G
.
BinaryCrossEntropyGrad
(),
'desc_inputs'
:
[
Tensor
([[
0.3
,
0.8
],
[
0.4
,
0.3
]],
mstype
.
float16
),
Tensor
([[
0.4
,
1.2
],
[
-
0.4
,
-
0.9
]],
mstype
.
float16
),
Tensor
(
0.85
,
mstype
.
float16
),
Tensor
([[
-
1.4
,
-
0.7
],
[
0.9
,
0.7
]],
mstype
.
float16
)],
'desc_bprop'
:
[],
'skip'
:
[
'backward'
]}),
]
test_case_array_ops
=
[
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录