Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
128dac7c
M
mindspore
项目概览
magicwindyyd
/
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看板
提交
128dac7c
编写于
6月 11, 2020
作者:
L
liuwenhao4
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Support dynamic broadcast reduce format with operator
上级
cf6dd99e
变更
88
隐藏空白更改
内联
并排
Showing
88 changed file
with
297 addition
and
519 deletion
+297
-519
mindspore/ops/_op_impl/tbe/acos.py
mindspore/ops/_op_impl/tbe/acos.py
+2
-4
mindspore/ops/_op_impl/tbe/acosh.py
mindspore/ops/_op_impl/tbe/acosh.py
+2
-4
mindspore/ops/_op_impl/tbe/add.py
mindspore/ops/_op_impl/tbe/add.py
+2
-7
mindspore/ops/_op_impl/tbe/asin.py
mindspore/ops/_op_impl/tbe/asin.py
+2
-2
mindspore/ops/_op_impl/tbe/asinh.py
mindspore/ops/_op_impl/tbe/asinh.py
+2
-2
mindspore/ops/_op_impl/tbe/atan.py
mindspore/ops/_op_impl/tbe/atan.py
+2
-4
mindspore/ops/_op_impl/tbe/atan2.py
mindspore/ops/_op_impl/tbe/atan2.py
+2
-4
mindspore/ops/_op_impl/tbe/atanh.py
mindspore/ops/_op_impl/tbe/atanh.py
+2
-4
mindspore/ops/_op_impl/tbe/batch_matmul.py
mindspore/ops/_op_impl/tbe/batch_matmul.py
+3
-4
mindspore/ops/_op_impl/tbe/bessel_i0e.py
mindspore/ops/_op_impl/tbe/bessel_i0e.py
+2
-2
mindspore/ops/_op_impl/tbe/bessel_i1e.py
mindspore/ops/_op_impl/tbe/bessel_i1e.py
+2
-2
mindspore/ops/_op_impl/tbe/bias_add.py
mindspore/ops/_op_impl/tbe/bias_add.py
+3
-3
mindspore/ops/_op_impl/tbe/bitwise_and.py
mindspore/ops/_op_impl/tbe/bitwise_and.py
+3
-2
mindspore/ops/_op_impl/tbe/bitwise_or.py
mindspore/ops/_op_impl/tbe/bitwise_or.py
+3
-2
mindspore/ops/_op_impl/tbe/bitwise_xor.py
mindspore/ops/_op_impl/tbe/bitwise_xor.py
+3
-2
mindspore/ops/_op_impl/tbe/bn_training_reduce.py
mindspore/ops/_op_impl/tbe/bn_training_reduce.py
+2
-2
mindspore/ops/_op_impl/tbe/bn_training_reduce_grad.py
mindspore/ops/_op_impl/tbe/bn_training_reduce_grad.py
+4
-4
mindspore/ops/_op_impl/tbe/bn_training_update_grad.py
mindspore/ops/_op_impl/tbe/bn_training_update_grad.py
+4
-4
mindspore/ops/_op_impl/tbe/bn_training_update_v2.py
mindspore/ops/_op_impl/tbe/bn_training_update_v2.py
+6
-6
mindspore/ops/_op_impl/tbe/ceil.py
mindspore/ops/_op_impl/tbe/ceil.py
+3
-2
mindspore/ops/_op_impl/tbe/check_valid.py
mindspore/ops/_op_impl/tbe/check_valid.py
+3
-2
mindspore/ops/_op_impl/tbe/clip_by_value.py
mindspore/ops/_op_impl/tbe/clip_by_value.py
+4
-6
mindspore/ops/_op_impl/tbe/concat.py
mindspore/ops/_op_impl/tbe/concat.py
+1
-22
mindspore/ops/_op_impl/tbe/conv2d.py
mindspore/ops/_op_impl/tbe/conv2d.py
+2
-0
mindspore/ops/_op_impl/tbe/cos.py
mindspore/ops/_op_impl/tbe/cos.py
+2
-4
mindspore/ops/_op_impl/tbe/cosh.py
mindspore/ops/_op_impl/tbe/cosh.py
+2
-2
mindspore/ops/_op_impl/tbe/div.py
mindspore/ops/_op_impl/tbe/div.py
+6
-10
mindspore/ops/_op_impl/tbe/div_no_nan.py
mindspore/ops/_op_impl/tbe/div_no_nan.py
+6
-10
mindspore/ops/_op_impl/tbe/dropout_do_mask.py
mindspore/ops/_op_impl/tbe/dropout_do_mask.py
+1
-2
mindspore/ops/_op_impl/tbe/elu.py
mindspore/ops/_op_impl/tbe/elu.py
+2
-2
mindspore/ops/_op_impl/tbe/equal.py
mindspore/ops/_op_impl/tbe/equal.py
+6
-10
mindspore/ops/_op_impl/tbe/erf.py
mindspore/ops/_op_impl/tbe/erf.py
+2
-2
mindspore/ops/_op_impl/tbe/erfc.py
mindspore/ops/_op_impl/tbe/erfc.py
+2
-2
mindspore/ops/_op_impl/tbe/exp.py
mindspore/ops/_op_impl/tbe/exp.py
+3
-4
mindspore/ops/_op_impl/tbe/expm1.py
mindspore/ops/_op_impl/tbe/expm1.py
+2
-2
mindspore/ops/_op_impl/tbe/floor.py
mindspore/ops/_op_impl/tbe/floor.py
+3
-2
mindspore/ops/_op_impl/tbe/floor_div.py
mindspore/ops/_op_impl/tbe/floor_div.py
+6
-5
mindspore/ops/_op_impl/tbe/floor_mod.py
mindspore/ops/_op_impl/tbe/floor_mod.py
+4
-3
mindspore/ops/_op_impl/tbe/fused_mul_add.py
mindspore/ops/_op_impl/tbe/fused_mul_add.py
+1
-15
mindspore/ops/_op_impl/tbe/gelu.py
mindspore/ops/_op_impl/tbe/gelu.py
+3
-8
mindspore/ops/_op_impl/tbe/greater.py
mindspore/ops/_op_impl/tbe/greater.py
+6
-10
mindspore/ops/_op_impl/tbe/greater_equal.py
mindspore/ops/_op_impl/tbe/greater_equal.py
+6
-10
mindspore/ops/_op_impl/tbe/inv.py
mindspore/ops/_op_impl/tbe/inv.py
+4
-3
mindspore/ops/_op_impl/tbe/inv_grad.py
mindspore/ops/_op_impl/tbe/inv_grad.py
+5
-4
mindspore/ops/_op_impl/tbe/invert.py
mindspore/ops/_op_impl/tbe/invert.py
+3
-2
mindspore/ops/_op_impl/tbe/layer_norm.py
mindspore/ops/_op_impl/tbe/layer_norm.py
+4
-12
mindspore/ops/_op_impl/tbe/layer_norm_beta_gamma_backprop.py
mindspore/ops/_op_impl/tbe/layer_norm_beta_gamma_backprop.py
+4
-12
mindspore/ops/_op_impl/tbe/layer_norm_x_backprop.py
mindspore/ops/_op_impl/tbe/layer_norm_x_backprop.py
+4
-12
mindspore/ops/_op_impl/tbe/less.py
mindspore/ops/_op_impl/tbe/less.py
+6
-10
mindspore/ops/_op_impl/tbe/less_equal.py
mindspore/ops/_op_impl/tbe/less_equal.py
+6
-10
mindspore/ops/_op_impl/tbe/log.py
mindspore/ops/_op_impl/tbe/log.py
+3
-4
mindspore/ops/_op_impl/tbe/log1p.py
mindspore/ops/_op_impl/tbe/log1p.py
+3
-4
mindspore/ops/_op_impl/tbe/logical_and.py
mindspore/ops/_op_impl/tbe/logical_and.py
+2
-4
mindspore/ops/_op_impl/tbe/logical_not.py
mindspore/ops/_op_impl/tbe/logical_not.py
+2
-4
mindspore/ops/_op_impl/tbe/logical_or.py
mindspore/ops/_op_impl/tbe/logical_or.py
+2
-4
mindspore/ops/_op_impl/tbe/maximum.py
mindspore/ops/_op_impl/tbe/maximum.py
+4
-6
mindspore/ops/_op_impl/tbe/maximum_grad.py
mindspore/ops/_op_impl/tbe/maximum_grad.py
+7
-12
mindspore/ops/_op_impl/tbe/minimum.py
mindspore/ops/_op_impl/tbe/minimum.py
+4
-6
mindspore/ops/_op_impl/tbe/minimum_grad.py
mindspore/ops/_op_impl/tbe/minimum_grad.py
+7
-12
mindspore/ops/_op_impl/tbe/neg.py
mindspore/ops/_op_impl/tbe/neg.py
+4
-8
mindspore/ops/_op_impl/tbe/not_equal.py
mindspore/ops/_op_impl/tbe/not_equal.py
+6
-10
mindspore/ops/_op_impl/tbe/ones_like.py
mindspore/ops/_op_impl/tbe/ones_like.py
+6
-5
mindspore/ops/_op_impl/tbe/pow.py
mindspore/ops/_op_impl/tbe/pow.py
+6
-5
mindspore/ops/_op_impl/tbe/reciprocal.py
mindspore/ops/_op_impl/tbe/reciprocal.py
+1
-6
mindspore/ops/_op_impl/tbe/reduce_all.py
mindspore/ops/_op_impl/tbe/reduce_all.py
+2
-2
mindspore/ops/_op_impl/tbe/reduce_max.py
mindspore/ops/_op_impl/tbe/reduce_max.py
+7
-6
mindspore/ops/_op_impl/tbe/reduce_min.py
mindspore/ops/_op_impl/tbe/reduce_min.py
+5
-8
mindspore/ops/_op_impl/tbe/reduce_prod.py
mindspore/ops/_op_impl/tbe/reduce_prod.py
+5
-8
mindspore/ops/_op_impl/tbe/reduce_sum.py
mindspore/ops/_op_impl/tbe/reduce_sum.py
+3
-2
mindspore/ops/_op_impl/tbe/relu.py
mindspore/ops/_op_impl/tbe/relu.py
+5
-9
mindspore/ops/_op_impl/tbe/relu6.py
mindspore/ops/_op_impl/tbe/relu6.py
+4
-6
mindspore/ops/_op_impl/tbe/relu_grad.py
mindspore/ops/_op_impl/tbe/relu_grad.py
+6
-10
mindspore/ops/_op_impl/tbe/round.py
mindspore/ops/_op_impl/tbe/round.py
+4
-6
mindspore/ops/_op_impl/tbe/rsqrt.py
mindspore/ops/_op_impl/tbe/rsqrt.py
+3
-8
mindspore/ops/_op_impl/tbe/select.py
mindspore/ops/_op_impl/tbe/select.py
+1
-10
mindspore/ops/_op_impl/tbe/sigmoid.py
mindspore/ops/_op_impl/tbe/sigmoid.py
+3
-10
mindspore/ops/_op_impl/tbe/sign.py
mindspore/ops/_op_impl/tbe/sign.py
+3
-3
mindspore/ops/_op_impl/tbe/sin.py
mindspore/ops/_op_impl/tbe/sin.py
+2
-4
mindspore/ops/_op_impl/tbe/sinh.py
mindspore/ops/_op_impl/tbe/sinh.py
+2
-2
mindspore/ops/_op_impl/tbe/softmax_grad_ext.py
mindspore/ops/_op_impl/tbe/softmax_grad_ext.py
+2
-8
mindspore/ops/_op_impl/tbe/softplus.py
mindspore/ops/_op_impl/tbe/softplus.py
+2
-2
mindspore/ops/_op_impl/tbe/softplus_grad.py
mindspore/ops/_op_impl/tbe/softplus_grad.py
+2
-2
mindspore/ops/_op_impl/tbe/split_d.py
mindspore/ops/_op_impl/tbe/split_d.py
+1
-22
mindspore/ops/_op_impl/tbe/sqrt.py
mindspore/ops/_op_impl/tbe/sqrt.py
+3
-6
mindspore/ops/_op_impl/tbe/square.py
mindspore/ops/_op_impl/tbe/square.py
+4
-9
mindspore/ops/_op_impl/tbe/sub.py
mindspore/ops/_op_impl/tbe/sub.py
+4
-6
mindspore/ops/_op_impl/tbe/tanh.py
mindspore/ops/_op_impl/tbe/tanh.py
+3
-6
mindspore/ops/_op_impl/tbe/unsorted_segment_sum.py
mindspore/ops/_op_impl/tbe/unsorted_segment_sum.py
+1
-10
未找到文件。
mindspore/ops/_op_impl/tbe/acos.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ acos_op_info = TBERegOp("ACos") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/acosh.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ acosh_op_info = TBERegOp("Acosh") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/add.py
浏览文件 @
128dac7c
...
...
@@ -23,16 +23,11 @@ add_op_info = TBERegOp("Add") \
.
compute_cost
(
10
)
\
.
kernel_name
(
"add"
)
\
.
partial_flag
(
True
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/asin.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ asin_op_info = TBERegOp("Asin") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/asinh.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ asinh_op_info = TBERegOp("Asinh") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/atan.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ atan_op_info = TBERegOp("Atan") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/atan2.py
浏览文件 @
128dac7c
...
...
@@ -27,10 +27,8 @@ atan2_op_info = TBERegOp("Atan2") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/atanh.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ atanh_op_info = TBERegOp("Atanh") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/batch_matmul.py
浏览文件 @
128dac7c
...
...
@@ -30,10 +30,9 @@ batch_matmul_op_info = TBERegOp("BatchMatMul") \
.
input
(
2
,
"bias"
,
False
,
"optional"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
,
DataType
.
F16_Default
,
DataType
.
F16_FracNZ
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bessel_i0e.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ bessel_i0e_op_info = TBERegOp("BesselI0e") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bessel_i1e.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ bessel_i1e_op_info = TBERegOp("BesselI1e") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bias_add.py
浏览文件 @
128dac7c
...
...
@@ -28,9 +28,9 @@ bias_add_grad_op_info = TBERegOp("BiasAdd") \
.
input
(
1
,
"bias"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
I32_
Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
I32_
None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bitwise_and.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,9 @@ bitwise_and_op_info = TBERegOp("BitwiseAnd") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I16_Default
,
DataType
.
I16_Default
,
DataType
.
I16_Default
)
\
.
dtype_format
(
DataType
.
U16_Default
,
DataType
.
U16_Default
,
DataType
.
U16_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I16_None
,
DataType
.
I16_None
,
DataType
.
I16_None
)
\
.
dtype_format
(
DataType
.
U16_None
,
DataType
.
U16_None
,
DataType
.
U16_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bitwise_or.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,9 @@ bitwise_or_op_info = TBERegOp("BitwiseOr") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I16_Default
,
DataType
.
I16_Default
,
DataType
.
I16_Default
)
\
.
dtype_format
(
DataType
.
U16_Default
,
DataType
.
U16_Default
,
DataType
.
U16_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I16_None
,
DataType
.
I16_None
,
DataType
.
I16_None
)
\
.
dtype_format
(
DataType
.
U16_None
,
DataType
.
U16_None
,
DataType
.
U16_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bitwise_xor.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,9 @@ bitwise_xor_op_info = TBERegOp("BitwiseXor") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I16_Default
,
DataType
.
I16_Default
,
DataType
.
I16_Default
)
\
.
dtype_format
(
DataType
.
U16_Default
,
DataType
.
U16_Default
,
DataType
.
U16_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I16_None
,
DataType
.
I16_None
,
DataType
.
I16_None
)
\
.
dtype_format
(
DataType
.
U16_None
,
DataType
.
U16_None
,
DataType
.
U16_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bn_training_reduce.py
浏览文件 @
128dac7c
...
...
@@ -27,8 +27,8 @@ bn_training_reduce_op_info = TBERegOp("BNTrainingReduce") \
.
output
(
0
,
"sum"
,
False
,
"required"
,
"all"
)
\
.
output
(
1
,
"square_sum"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bn_training_reduce_grad.py
浏览文件 @
128dac7c
...
...
@@ -33,10 +33,10 @@ bn_training_reduce_grad_op_info = TBERegOp("BNTrainingReduceGrad") \
.
input
(
6
,
"batch_variance"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
,
reshape_type
=
"NC"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bn_training_update_grad.py
浏览文件 @
128dac7c
...
...
@@ -31,10 +31,10 @@ bn_training_update_grad_op_info = TBERegOp("BNTrainingUpdateGrad") \
.
output
(
0
,
"diff_scale"
,
False
,
"required"
,
"all"
)
\
.
output
(
1
,
"diff_offset"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/bn_training_update_v2.py
浏览文件 @
128dac7c
...
...
@@ -33,12 +33,12 @@ bn_training_update_v2_op_info = TBERegOp("BNTrainingUpdateV2") \
.
output
(
1
,
"batch_mean"
,
False
,
"required"
,
"all"
)
\
.
output
(
2
,
"batch_variance"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F16_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F16_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/ceil.py
浏览文件 @
128dac7c
...
...
@@ -25,8 +25,9 @@ ceil_op_info = TBERegOp("Ceil") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/check_valid.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,9 @@ check_valid_op_info = TBERegOp("CheckValid") \
.
input
(
0
,
"bbox_tensor"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"img_tas"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"valid_tensor"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/clip_by_value.py
浏览文件 @
128dac7c
...
...
@@ -28,12 +28,10 @@ clip_by_value_op_info = TBERegOp("ClipByValue") \
.
input
(
1
,
"clip_value_min"
,
False
,
"required"
,
"all"
)
\
.
input
(
2
,
"clip_value_max"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
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
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/concat.py
浏览文件 @
128dac7c
...
...
@@ -27,28 +27,7 @@ concat_op_info = TBERegOp("Concat") \
.
input
(
0
,
"input_values"
,
False
,
"dynamic"
,
"all"
)
\
.
output
(
0
,
"output_data"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
)
\
.
dtype_format
(
DataType
.
I16_Default
,
DataType
.
I16_Default
)
\
.
dtype_format
(
DataType
.
I16_5HD
,
DataType
.
I16_5HD
)
\
.
dtype_format
(
DataType
.
U16_Default
,
DataType
.
U16_Default
)
\
.
dtype_format
(
DataType
.
U16_5HD
,
DataType
.
U16_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
U32_Default
,
DataType
.
U32_Default
)
\
.
dtype_format
(
DataType
.
U32_5HD
,
DataType
.
U32_5HD
)
\
.
dtype_format
(
DataType
.
I64_Default
,
DataType
.
I64_Default
)
\
.
dtype_format
(
DataType
.
I64_5HD
,
DataType
.
I64_5HD
)
\
.
dtype_format
(
DataType
.
U64_Default
,
DataType
.
U64_Default
)
\
.
dtype_format
(
DataType
.
U64_5HD
,
DataType
.
U64_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/conv2d.py
浏览文件 @
128dac7c
...
...
@@ -33,7 +33,9 @@ conv2d_op_info = TBERegOp("Conv2D") \
.
input
(
2
,
"bias"
,
False
,
"optional"
,
"all"
)
\
.
input
(
3
,
"offset_w"
,
False
,
"optional"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
I8_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I32_None
,
DataType
.
I8_None
,
DataType
.
I32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/cos.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ cos_op_info = TBERegOp("Cos") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/cosh.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ cosh_op_info = TBERegOp("Cosh") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/div.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ div_op_info = TBERegOp("Div") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
U8_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/div_no_nan.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ div_no_nan_op_info = TBERegOp("DivNoNan") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
U8_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/dropout_do_mask.py
浏览文件 @
128dac7c
...
...
@@ -28,8 +28,7 @@ drop_out_do_mask_op_info = TBERegOp("DropoutDoMask") \
.
input
(
2
,
"keep_prob"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
U8_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
U8_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/elu.py
浏览文件 @
128dac7c
...
...
@@ -27,8 +27,8 @@ elu_op_info = TBERegOp("Elu") \
.
attr
(
"alpha"
,
"optional"
,
"float"
,
"all"
,
"1.0"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/equal.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ equal_op_info = TBERegOp("Equal") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/erf.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ erf_op_info = TBERegOp("Erf") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/erfc.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ erfc_op_info = TBERegOp("Erfc") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/exp.py
浏览文件 @
128dac7c
...
...
@@ -25,10 +25,9 @@ exp_op_info = TBERegOp("Exp") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/expm1.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ expm1_op_info = TBERegOp("Expm1") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/floor.py
浏览文件 @
128dac7c
...
...
@@ -25,8 +25,9 @@ floor_op_info = TBERegOp("Floor") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/floor_div.py
浏览文件 @
128dac7c
...
...
@@ -26,11 +26,12 @@ floordiv_op_info = TBERegOp("FloorDiv") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/floor_mod.py
浏览文件 @
128dac7c
...
...
@@ -26,9 +26,10 @@ floor_mod_op_info = TBERegOp("FloorMod") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/fused_mul_add.py
浏览文件 @
128dac7c
...
...
@@ -28,21 +28,7 @@ fused_mul_add_op_info = TBERegOp("FusedMulAdd") \
.
input
(
2
,
"x3"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
I32_FracZ
,
DataType
.
I32_FracZ
,
DataType
.
I32_FracZ
,
DataType
.
I32_FracZ
)
\
.
dtype_format
(
DataType
.
I32_FracNZ
,
DataType
.
I32_FracNZ
,
DataType
.
I32_FracNZ
,
DataType
.
I32_FracNZ
)
\
.
dtype_format
(
DataType
.
I32_C1HWNCoC0
,
DataType
.
I32_C1HWNCoC0
,
DataType
.
I32_C1HWNCoC0
,
DataType
.
I32_C1HWNCoC0
)
\
.
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
.
F16_FracZ
,
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
)
\
.
dtype_format
(
DataType
.
F16_C1HWNCoC0
,
DataType
.
F16_C1HWNCoC0
,
DataType
.
F16_C1HWNCoC0
,
DataType
.
F16_C1HWNCoC0
)
\
.
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
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
)
\
.
dtype_format
(
DataType
.
F32_C1HWNCoC0
,
DataType
.
F32_C1HWNCoC0
,
DataType
.
F32_C1HWNCoC0
,
DataType
.
F32_C1HWNCoC0
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/gelu.py
浏览文件 @
128dac7c
...
...
@@ -25,14 +25,9 @@ gelu_op_info = TBERegOp("Gelu") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/greater.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ greater_op_info = TBERegOp("Greater") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/greater_equal.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ greater_equal_op_info = TBERegOp("GreaterEqual") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/inv.py
浏览文件 @
128dac7c
...
...
@@ -25,9 +25,10 @@ inv_op_info = TBERegOp("Inv") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/inv_grad.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,11 @@ inv_grad_op_info = TBERegOp("InvGrad") \
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"grad"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/invert.py
浏览文件 @
128dac7c
...
...
@@ -25,8 +25,9 @@ invert_op_info = TBERegOp("Invert") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I16_Default
,
DataType
.
I16_Default
)
\
.
dtype_format
(
DataType
.
U16_Default
,
DataType
.
U16_Default
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
I16_None
,
DataType
.
I16_None
)
\
.
dtype_format
(
DataType
.
U16_None
,
DataType
.
U16_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/layer_norm.py
浏览文件 @
128dac7c
...
...
@@ -33,18 +33,10 @@ layer_norm_op_info = TBERegOp("LayerNorm") \
.
output
(
1
,
"mean"
,
False
,
"required"
,
"all"
)
\
.
output
(
2
,
"variance"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
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
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_FracNZ
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
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
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_FracNZ
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/layer_norm_beta_gamma_backprop.py
浏览文件 @
128dac7c
...
...
@@ -31,18 +31,10 @@ layer_norm_beta_gamma_backprop_op_info = TBERegOp("LayerNormBetaGammaBackprop")
.
output
(
0
,
"pd_gamma"
,
False
,
"required"
,
"all"
)
\
.
output
(
1
,
"pd_beta"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
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
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/layer_norm_x_backprop.py
浏览文件 @
128dac7c
...
...
@@ -30,18 +30,10 @@ layer_norm_x_backprop_op_info = TBERegOp("LayerNormXBackprop") \
.
input
(
4
,
"gamma"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"pd_x"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
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
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_FracNZ
)
\
.
dtype_format
(
DataType
.
F32_Default
,
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
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_FracNZ
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/less.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ less_op_info = TBERegOp("Less") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/less_equal.py
浏览文件 @
128dac7c
...
...
@@ -28,16 +28,12 @@ less_equal_op_info = TBERegOp("LessEqual") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/log.py
浏览文件 @
128dac7c
...
...
@@ -25,10 +25,9 @@ log_op_info = TBERegOp("Log") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/log1p.py
浏览文件 @
128dac7c
...
...
@@ -25,10 +25,9 @@ log1p_op_info = TBERegOp("Log1p") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/logical_and.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ logical_and_op_info = TBERegOp("LogicalAnd") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
BOOL_FracZ
,
DataType
.
BOOL_FracZ
,
DataType
.
BOOL_FracZ
)
\
.
dtype_format
(
DataType
.
BOOL_C1HWNCoC0
,
DataType
.
BOOL_C1HWNCoC0
,
DataType
.
BOOL_C1HWNCoC0
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
BOOL_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
BOOL_None
,
DataType
.
BOOL_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/logical_not.py
浏览文件 @
128dac7c
...
...
@@ -25,10 +25,8 @@ logical_not_op_info = TBERegOp("LogicalNot") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
BOOL_FracZ
,
DataType
.
BOOL_FracZ
)
\
.
dtype_format
(
DataType
.
BOOL_C1HWNCoC0
,
DataType
.
BOOL_C1HWNCoC0
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
BOOL_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/logical_or.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ logical_or_op_info = TBERegOp("LogicalOr") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
BOOL_FracZ
,
DataType
.
BOOL_FracZ
,
DataType
.
BOOL_FracZ
)
\
.
dtype_format
(
DataType
.
BOOL_C1HWNCoC0
,
DataType
.
BOOL_C1HWNCoC0
,
DataType
.
BOOL_C1HWNCoC0
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
BOOL_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
BOOL_None
,
DataType
.
BOOL_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/maximum.py
浏览文件 @
128dac7c
...
...
@@ -26,12 +26,10 @@ maximum_op_info = TBERegOp("Maximum") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/maximum_grad.py
浏览文件 @
128dac7c
...
...
@@ -30,18 +30,13 @@ maximum_grad_op_info = TBERegOp("MaximumGrad") \
.
input
(
2
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y1"
,
False
,
"required"
,
"all"
)
\
.
output
(
1
,
"y2"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
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
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/minimum.py
浏览文件 @
128dac7c
...
...
@@ -27,12 +27,10 @@ minimum_op_info = TBERegOp("Minimum") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/minimum_grad.py
浏览文件 @
128dac7c
...
...
@@ -30,18 +30,13 @@ minimum_grad_op_info = TBERegOp("MinimumGrad") \
.
input
(
2
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y1"
,
False
,
"required"
,
"all"
)
\
.
output
(
1
,
"y2"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
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
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/neg.py
浏览文件 @
128dac7c
...
...
@@ -25,14 +25,10 @@ neg_op_info = TBERegOp("Neg") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/not_equal.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ not_equal_op_info = TBERegOp("NotEqual") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
BOOL_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
BOOL_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/ones_like.py
浏览文件 @
128dac7c
...
...
@@ -25,11 +25,12 @@ ones_like_op_info = TBERegOp("OnesLike") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/pow.py
浏览文件 @
128dac7c
...
...
@@ -26,11 +26,12 @@ pow_op_info = TBERegOp("Pow") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/reciprocal.py
浏览文件 @
128dac7c
...
...
@@ -26,12 +26,7 @@ reciprocal_op_info = TBERegOp("Reciprocal") \
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_NHWC
,
DataType
.
F16_NHWC
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_NHWC
,
DataType
.
F32_NHWC
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/reduce_all.py
浏览文件 @
128dac7c
...
...
@@ -27,8 +27,8 @@ reduce_all_op_info = TBERegOp("ReduceAll") \
.
attr
(
"keep_dims"
,
"optional"
,
"bool"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
BOOL_
FracZ
,
DataType
.
BOOL_FracZ
)
\
.
op_pattern
(
"reduce"
)
\
.
dtype_format
(
DataType
.
BOOL_
None
,
DataType
.
BOOL_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/reduce_max.py
浏览文件 @
128dac7c
...
...
@@ -27,12 +27,13 @@ reduce_max_d_op_info = TBERegOp("ReduceMax") \
.
attr
(
"keep_dims"
,
"optional"
,
"bool"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"reduce"
)
\
.
dtype_format
(
DataType
.
BOOL_None
,
DataType
.
BOOL_None
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/reduce_min.py
浏览文件 @
128dac7c
...
...
@@ -27,14 +27,11 @@ reduce_min_op_info = TBERegOp("ReduceMin") \
.
attr
(
"keep_dims"
,
"required"
,
"bool"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_FracZ
,
DataType
.
I8_FracZ
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_FracZ
,
DataType
.
U8_FracZ
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
op_pattern
(
"reduce"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/reduce_prod.py
浏览文件 @
128dac7c
...
...
@@ -27,14 +27,11 @@ reduce_prod_op_info = TBERegOp("ReduceProd") \
.
attr
(
"keep_dims"
,
"optional"
,
"bool"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_FracZ
,
DataType
.
I8_FracZ
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_FracZ
,
DataType
.
U8_FracZ
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
op_pattern
(
"reduce"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/reduce_sum.py
浏览文件 @
128dac7c
...
...
@@ -27,8 +27,9 @@ reduce_sum_op_info = TBERegOp("ReduceSum") \
.
attr
(
"keep_dims"
,
"optional"
,
"bool"
,
"all"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
op_pattern
(
"reduce"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/relu.py
浏览文件 @
128dac7c
...
...
@@ -25,15 +25,11 @@ relu_op_info = TBERegOp("ReLU") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/relu6.py
浏览文件 @
128dac7c
...
...
@@ -25,12 +25,10 @@ relu6_op_info = TBERegOp("ReLU6") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/relu_grad.py
浏览文件 @
128dac7c
...
...
@@ -26,16 +26,12 @@ relugrad_op_info = TBERegOp("ReluGrad") \
.
input
(
0
,
"gradients"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"features"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"backprops"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
U8_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I8_None
,
DataType
.
I8_None
,
DataType
.
I8_None
)
\
.
dtype_format
(
DataType
.
U8_None
,
DataType
.
U8_None
,
DataType
.
U8_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/round.py
浏览文件 @
128dac7c
...
...
@@ -25,12 +25,10 @@ round_op_info = TBERegOp("Round") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/rsqrt.py
浏览文件 @
128dac7c
...
...
@@ -25,14 +25,9 @@ rsqrt_op_info = TBERegOp("Rsqrt") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F16_C1HWNCoC0
,
DataType
.
F16_C1HWNCoC0
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
dtype_format
(
DataType
.
F32_C1HWNCoC0
,
DataType
.
F32_C1HWNCoC0
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/select.py
浏览文件 @
128dac7c
...
...
@@ -28,16 +28,7 @@ select_op_info = TBERegOp("Select") \
.
input
(
2
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
I8_5HD
,
DataType
.
I8_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
U8_5HD
,
DataType
.
U8_5HD
,
DataType
.
U8_5HD
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
BOOL_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/sigmoid.py
浏览文件 @
128dac7c
...
...
@@ -25,16 +25,9 @@ sigmoid_op_info = TBERegOp("Sigmoid") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_FracZ
,
DataType
.
F16_FracZ
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
)
\
.
dtype_format
(
DataType
.
F16_C1HWNCoC0
,
DataType
.
F16_C1HWNCoC0
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracZ
,
DataType
.
F32_FracZ
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
)
\
.
dtype_format
(
DataType
.
F32_C1HWNCoC0
,
DataType
.
F32_C1HWNCoC0
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/sign.py
浏览文件 @
128dac7c
...
...
@@ -26,9 +26,9 @@ sign_op_info = TBERegOp("Sign") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
None
,
"required"
,
None
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
I32_
Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
dtype_format
(
DataType
.
I32_
None
,
DataType
.
I32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/sin.py
浏览文件 @
128dac7c
...
...
@@ -26,10 +26,8 @@ sin_op_info = TBERegOp("Sin") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/sinh.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ sinh_op_info = TBERegOp("Sinh") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_
5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/softmax_grad_ext.py
浏览文件 @
128dac7c
...
...
@@ -31,14 +31,8 @@ softmax_grad_ext_op_info = TBERegOp("SoftmaxGradExt") \
.
input
(
2
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
True
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
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
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/softplus.py
浏览文件 @
128dac7c
...
...
@@ -26,8 +26,8 @@ softplus_op_info = TBERegOp("Softplus") \
.
op_pattern
(
"formatAgnostic"
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/softplus_grad.py
浏览文件 @
128dac7c
...
...
@@ -27,8 +27,8 @@ softplus_grad_op_info = TBERegOp("SoftplusGrad") \
.
input
(
0
,
"gradients"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"features"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"backprops"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_
Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_
Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F16_
None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_
None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/split_d.py
浏览文件 @
128dac7c
...
...
@@ -28,28 +28,7 @@ split_d_op_info = TBERegOp("Split") \
.
input
(
0
,
"value"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"output"
,
False
,
"dynamic"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
BOOL_Default
,
DataType
.
BOOL_Default
)
\
.
dtype_format
(
DataType
.
BOOL_NHWC
,
DataType
.
BOOL_NHWC
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_NHWC
,
DataType
.
I8_NHWC
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_NHWC
,
DataType
.
U8_NHWC
)
\
.
dtype_format
(
DataType
.
I16_Default
,
DataType
.
I16_Default
)
\
.
dtype_format
(
DataType
.
I16_NHWC
,
DataType
.
I16_NHWC
)
\
.
dtype_format
(
DataType
.
U16_Default
,
DataType
.
U16_Default
)
\
.
dtype_format
(
DataType
.
U16_NHWC
,
DataType
.
U16_NHWC
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_NHWC
,
DataType
.
I32_NHWC
)
\
.
dtype_format
(
DataType
.
U32_Default
,
DataType
.
U32_Default
)
\
.
dtype_format
(
DataType
.
U32_NHWC
,
DataType
.
U32_NHWC
)
\
.
dtype_format
(
DataType
.
I64_Default
,
DataType
.
I64_Default
)
\
.
dtype_format
(
DataType
.
I64_NHWC
,
DataType
.
I64_NHWC
)
\
.
dtype_format
(
DataType
.
U64_Default
,
DataType
.
U64_Default
)
\
.
dtype_format
(
DataType
.
U64_NHWC
,
DataType
.
U64_NHWC
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_NHWC
,
DataType
.
F16_NHWC
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_NHWC
,
DataType
.
F32_NHWC
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/sqrt.py
浏览文件 @
128dac7c
...
...
@@ -25,12 +25,9 @@ sqrt_op_info = TBERegOp("Sqrt") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_NHWC
,
DataType
.
F16_NHWC
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_NHWC
,
DataType
.
F32_NHWC
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/square.py
浏览文件 @
128dac7c
...
...
@@ -25,15 +25,10 @@ square_op_info = TBERegOp("Square") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
I32_NHWC
,
DataType
.
I32_NHWC
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F16_NHWC
,
DataType
.
F16_NHWC
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_NHWC
,
DataType
.
F32_NHWC
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/sub.py
浏览文件 @
128dac7c
...
...
@@ -26,12 +26,10 @@ sub_op_info = TBERegOp("Sub") \
.
input
(
0
,
"x1"
,
False
,
"required"
,
"all"
)
\
.
input
(
1
,
"x2"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
op_pattern
(
"broadcast"
)
\
.
dtype_format
(
DataType
.
I32_None
,
DataType
.
I32_None
,
DataType
.
I32_None
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/tanh.py
浏览文件 @
128dac7c
...
...
@@ -25,12 +25,9 @@ tanh_op_info = TBERegOp("Tanh") \
.
partial_flag
(
True
)
\
.
input
(
0
,
"x"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
F32_FracNZ
,
DataType
.
F32_FracNZ
)
\
.
dtype_format
(
DataType
.
F16_FracNZ
,
DataType
.
F16_FracNZ
)
\
.
op_pattern
(
"formatAgnostic"
)
\
.
dtype_format
(
DataType
.
F16_None
,
DataType
.
F16_None
)
\
.
dtype_format
(
DataType
.
F32_None
,
DataType
.
F32_None
)
\
.
get_op_info
()
...
...
mindspore/ops/_op_impl/tbe/unsorted_segment_sum.py
浏览文件 @
128dac7c
...
...
@@ -28,16 +28,7 @@ unsorted_segment_sum_op_info = TBERegOp("UnsortedSegmentSum") \
.
input
(
1
,
"segment_ids"
,
False
,
"required"
,
"all"
)
\
.
output
(
0
,
"y"
,
False
,
"required"
,
"all"
)
\
.
op_pattern
(
"dynamicFormat"
)
\
.
dtype_format
(
DataType
.
I8_Default
,
DataType
.
I32_Default
,
DataType
.
I8_Default
)
\
.
dtype_format
(
DataType
.
I8_5HD
,
DataType
.
I32_5HD
,
DataType
.
I8_5HD
)
\
.
dtype_format
(
DataType
.
U8_Default
,
DataType
.
I32_Default
,
DataType
.
U8_Default
)
\
.
dtype_format
(
DataType
.
U8_5HD
,
DataType
.
I32_5HD
,
DataType
.
U8_5HD
)
\
.
dtype_format
(
DataType
.
I32_Default
,
DataType
.
I32_Default
,
DataType
.
I32_Default
)
\
.
dtype_format
(
DataType
.
I32_5HD
,
DataType
.
I32_5HD
,
DataType
.
I32_5HD
)
\
.
dtype_format
(
DataType
.
F16_Default
,
DataType
.
I32_Default
,
DataType
.
F16_Default
)
\
.
dtype_format
(
DataType
.
F16_5HD
,
DataType
.
I32_5HD
,
DataType
.
F16_5HD
)
\
.
dtype_format
(
DataType
.
F32_Default
,
DataType
.
I32_Default
,
DataType
.
F32_Default
)
\
.
dtype_format
(
DataType
.
F32_5HD
,
DataType
.
I32_5HD
,
DataType
.
F32_5HD
)
\
.
dtype_format
(
DataType
.
None_None
,
DataType
.
None_None
,
DataType
.
None_None
)
\
.
get_op_info
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录