!30700 Support dynamic shape for a few ops in Array, Math, Grad Array and Grad Math

Merge pull request !30700 from zhengzuohe/fill_stack_reshape
This commit is contained in:
i-robot 2022-03-14 05:02:23 +00:00 committed by Gitee
commit 53ff859661
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
41 changed files with 522 additions and 456 deletions

View File

@ -208,6 +208,8 @@ def get_bprop_squeeze(self):
def bprop(x, out, dout):
shapex = shape_op(x)
if is_shape_unknown(shapex):
shapex = dyn_shape_op(x)
return (reshape(dout, shapex),)
return bprop

View File

@ -43,7 +43,7 @@ to_array = P.TupleToArray()
real_div = P.RealDiv()
def dyn_binop_grad_common(x, y, dx, dy):
def dyn_binop_grad_common(x, y, dx, dy, shift=0):
"""
Common grad definition for binary operations when the input is dynamic shape.
@ -51,7 +51,12 @@ def dyn_binop_grad_common(x, y, dx, dy):
"""
shape_of_x = dyn_shape_op(x)
shape_of_y = dyn_shape_op(y)
rx, ry = DynamicBroadcastGradientArgs()(shape_of_x, shape_of_y)
broadcast_shape_of_x = shape_of_x
broadcast_shape_of_y = shape_of_y
if shift != 0:
broadcast_shape_of_x = shape_of_x[:-shift]
broadcast_shape_of_y = shape_of_y[:-shift]
rx, ry = DynamicBroadcastGradientArgs()(broadcast_shape_of_x, broadcast_shape_of_y)
dx = reduce_sum(dx, rx)
dy = reduce_sum(dy, ry)
reduce_dx = reshape(dx, shape_of_x)
@ -87,7 +92,7 @@ def binop_grad_common(x, y, dx, dy, shift=0):
# if input shape is the same as dout shape, do not need to reduce
reduce_dx = dx
reduce_dy = dy
if not (is_shape_unknown(shape_of_x) or is_shape_unknown(shape_of_y)):
if not (is_shape_unknown(broadcast_shape_of_x) or is_shape_unknown(broadcast_shape_of_y)):
rx = broadcast_gradient_args(broadcast_shape_of_x, broadcast_shape_of_y)
if rx[0]:
# if dx is scalar whose shape is (), do not need reduce
@ -108,7 +113,7 @@ def binop_grad_common(x, y, dx, dy, shift=0):
reduce_dy = reduce_sum_with_cast(dy, ())
return reduce_dx, reduce_dy
return dyn_binop_grad_common(x, y, dx, dy)
return dyn_binop_grad_common(x, y, dx, dy, shift)
def _dyn_reduced_shape(input_shape, axis):

View File

@ -1,20 +1,20 @@
0.1.1 MindSpore*1.7.0:Þ

0.1.1 MindSpore*1.7.0:Ü

bprop.43:xbprop.43:[CNode]44:1bprop.43:[CNode]44:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op21

bprop.24:xbprop.24:[CNode]25:1bprop.24:[CNode]25:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op8

bprop.43:ybprop.43:[CNode]45:3bprop.43:[CNode]45:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op22
bprop.24:ybprop.24:[CNode]26:3bprop.24:[CNode]26:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op9
<EFBFBD>
bprop.43:[CNode]44:1
bprop.43:[CNode]45:3bprop.43:[CNode]46:4bprop.43:[CNode]46:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op23bprop.43*
bprop.24:[CNode]25:1
bprop.24:[CNode]26:3bprop.24:[CNode]27:4bprop.24:[CNode]27:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op10bprop.24*
bprop.43:x*
bprop.24:x*
bprop.43:y*
bprop.43:out*
bprop.43:dout2
bprop.43:[CNode]46:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
bprop.24:y*
bprop.24:out*
bprop.24:dout2
bprop.24:[CNode]27:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,14 +1,14 @@
0.1.1 MindSpore*1.7.0:£
0.1.1 MindSpore*1.7.0:¢
<EFBFBD>
bprop.20:xbprop.20:[CNode]21:1bprop.20:[CNode]21:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op9
z
bprop.20:[CNode]21:1bprop.20:[CNode]22:3bprop.20:[CNode]22:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op10bprop.20*
bprop.18:xbprop.18:[CNode]19:1bprop.18:[CNode]19:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op7
y
bprop.18:[CNode]19:1bprop.18:[CNode]20:3bprop.18:[CNode]20:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op8bprop.18*
bprop.20:x*
bprop.20:out*
bprop.20:dout2
bprop.20:[CNode]22:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
bprop.18:x*
bprop.18:out*
bprop.18:dout2
bprop.18:[CNode]20:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7PbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -1,14 +1,14 @@
0.1.1 MindSpore*1.7.0:¤

0.1.1 MindSpore*1.7.0:£

bprop.24:xbprop.24:[CNode]25:1bprop.24:[CNode]25:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op11
bprop.22:xbprop.22:[CNode]23:1bprop.22:[CNode]23:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op9
z
bprop.24:[CNode]25:1bprop.24:[CNode]26:3bprop.24:[CNode]26:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op12bprop.24*
bprop.22:[CNode]23:1bprop.22:[CNode]24:3bprop.22:[CNode]24:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op10bprop.22*
bprop.24:x*
bprop.24:out*
bprop.24:dout2
bprop.24:[CNode]26:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
bprop.22:x*
bprop.22:out*
bprop.22:dout2
bprop.22:[CNode]24:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7Pb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,20 +1,20 @@
0.1.1 MindSpore*1.7.0:Þ

0.1.1 MindSpore*1.7.0:Û

bprop.29:xbprop.29:[CNode]30:1bprop.29:[CNode]30:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op13

bprop.10:xbprop.10:[CNode]11:1bprop.10:[CNode]11:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op0

bprop.29:ybprop.29:[CNode]31:3bprop.29:[CNode]31:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op14
<EFBFBD>
bprop.29:[CNode]30:1
bprop.29:[CNode]31:3bprop.29:[CNode]32:4bprop.29:[CNode]32:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op15bprop.29*
bprop.10:ybprop.10:[CNode]12:3bprop.10:[CNode]12:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op1
<EFBFBD>
bprop.10:[CNode]11:1
bprop.10:[CNode]12:3bprop.10:[CNode]13:4bprop.10:[CNode]13:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op2bprop.10*
bprop.29:x*
bprop.10:x*
bprop.29:y*
bprop.29:out*
bprop.29:dout2
bprop.29:[CNode]32:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
bprop.10:y*
bprop.10:out*
bprop.10:dout2
bprop.10:[CNode]13:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,20 +1,20 @@
0.1.1 MindSpore*1.7.0:Þ

0.1.1 MindSpore*1.7.0:Û

bprop.34:xbprop.34:[CNode]35:1bprop.34:[CNode]35:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op16

bprop.15:xbprop.15:[CNode]16:1bprop.15:[CNode]16:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op3

bprop.34:ybprop.34:[CNode]36:3bprop.34:[CNode]36:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op17
<EFBFBD>
bprop.34:[CNode]35:1
bprop.34:[CNode]36:3bprop.34:[CNode]37:4bprop.34:[CNode]37:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op18bprop.34*
bprop.15:ybprop.15:[CNode]17:3bprop.15:[CNode]17:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op4
<EFBFBD>
bprop.15:[CNode]16:1
bprop.15:[CNode]17:3bprop.15:[CNode]18:4bprop.15:[CNode]18:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op5bprop.15*
bprop.34:x*
bprop.15:x*
bprop.34:y*
bprop.34:out*
bprop.34:dout2
bprop.34:[CNode]37:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
bprop.15:y*
bprop.15:out*
bprop.15:dout2
bprop.15:[CNode]18:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:¶

bprop.137:xbprop.137:[CNode]138:1bprop.137:[CNode]138:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op76
bprop.120:xbprop.120:[CNode]121:1bprop.120:[CNode]121:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op62

bprop.137:[CNode]138:1bprop.137:[CNode]139:3bprop.137:[CNode]139:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op77 bprop.137*
bprop.137:x*
bprop.137:out*
bprop.137:dout2
bprop.137:[CNode]139:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPb&
bprop.120:[CNode]121:1bprop.120:[CNode]122:3bprop.120:[CNode]122:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op63 bprop.120*
bprop.120:x*
bprop.120:out*
bprop.120:dout2
bprop.120:[CNode]122:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7Pb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:¶

bprop.145:xbprop.145:[CNode]146:1bprop.145:[CNode]146:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op80
bprop.128:xbprop.128:[CNode]129:1bprop.128:[CNode]129:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op66

bprop.145:[CNode]146:1bprop.145:[CNode]147:3bprop.145:[CNode]147:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op81 bprop.145*
bprop.145:x*
bprop.145:out*
bprop.145:dout2
bprop.145:[CNode]147:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
bprop.128:[CNode]129:1bprop.128:[CNode]130:3bprop.128:[CNode]130:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op67 bprop.128*
bprop.128:x*
bprop.128:out*
bprop.128:dout2
bprop.128:[CNode]130:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7PbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -2,19 +2,19 @@
0.1.1 MindSpore*1.7.0:Ţ

bprop.81:xbprop.81:[CNode]82:1bprop.81:[CNode]82:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op42
bprop.52:xbprop.52:[CNode]53:1bprop.52:[CNode]53:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op21

bprop.81:ybprop.81:[CNode]83:3bprop.81:[CNode]83:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op43
bprop.52:ybprop.52:[CNode]54:3bprop.52:[CNode]54:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op22
<EFBFBD>
bprop.81:[CNode]82:1
bprop.81:[CNode]83:3bprop.81:[CNode]84:4bprop.81:[CNode]84:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op44bprop.81*
bprop.52:[CNode]53:1
bprop.52:[CNode]54:3bprop.52:[CNode]55:4bprop.52:[CNode]55:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op23bprop.52*
bprop.81:x*
bprop.52:x*
bprop.81:y*
bprop.81:out*
bprop.81:dout2
bprop.81:[CNode]84:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
bprop.52:y*
bprop.52:out*
bprop.52:dout2
bprop.52:[CNode]55:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -1,16 +1,16 @@
0.1.1 MindSpore*1.7.0:÷

bprop.172:xbprop.172:[CNode]173:1bprop.172:[CNode]173:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op96
bprop.111:xbprop.111:[CNode]112:1bprop.111:[CNode]112:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op51

bprop.172:ybprop.172:[CNode]174:3bprop.172:[CNode]174:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op97
bprop.111:ybprop.111:[CNode]113:3bprop.111:[CNode]113:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op52
˜
bprop.172:[CNode]173:1
bprop.172:[CNode]174:3bprop.172:[CNode]175:4bprop.172:[CNode]175:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op98 bprop.172*
bprop.172:x*
bprop.172:y*
bprop.172:out*
bprop.172:dout2
bprop.172:[CNode]175:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
bprop.111:[CNode]112:1
bprop.111:[CNode]113:3bprop.111:[CNode]114:4bprop.111:[CNode]114:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op53 bprop.111*
bprop.111:x*
bprop.111:y*
bprop.111:out*
bprop.111:dout2
bprop.111:[CNode]114:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -2,19 +2,19 @@
0.1.1 MindSpore*1.7.0:Ţ

bprop.96:xbprop.96:[CNode]97:1bprop.96:[CNode]97:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op51
bprop.67:xbprop.67:[CNode]68:1bprop.67:[CNode]68:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op30

bprop.96:ybprop.96:[CNode]98:3bprop.96:[CNode]98:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op52
bprop.67:ybprop.67:[CNode]69:3bprop.67:[CNode]69:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op31
<EFBFBD>
bprop.96:[CNode]97:1
bprop.96:[CNode]98:3bprop.96:[CNode]99:4bprop.96:[CNode]99:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op53bprop.96*
bprop.67:[CNode]68:1
bprop.67:[CNode]69:3bprop.67:[CNode]70:4bprop.67:[CNode]70:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op32bprop.67*
bprop.96:x*
bprop.67:x*
bprop.96:y*
bprop.96:out*
bprop.96:dout2
bprop.96:[CNode]99:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
bprop.67:y*
bprop.67:out*
bprop.67:dout2
bprop.67:[CNode]70:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -2,19 +2,19 @@
0.1.1 MindSpore*1.7.0:Ţ

bprop.91:xbprop.91:[CNode]92:1bprop.91:[CNode]92:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op48
bprop.62:xbprop.62:[CNode]63:1bprop.62:[CNode]63:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op27

bprop.91:ybprop.91:[CNode]93:3bprop.91:[CNode]93:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op49
bprop.62:ybprop.62:[CNode]64:3bprop.62:[CNode]64:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op28
<EFBFBD>
bprop.91:[CNode]92:1
bprop.91:[CNode]93:3bprop.91:[CNode]94:4bprop.91:[CNode]94:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op50bprop.91*
bprop.62:[CNode]63:1
bprop.62:[CNode]64:3bprop.62:[CNode]65:4bprop.62:[CNode]65:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op29bprop.62*
bprop.91:x*
bprop.62:x*
bprop.91:y*
bprop.91:out*
bprop.91:dout2
bprop.91:[CNode]94:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
bprop.62:y*
bprop.62:out*
bprop.62:dout2
bprop.62:[CNode]65:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -1,9 +1,9 @@
0.1.1 MindSpore*1.7.0:ü
m
bprop.5:doutbprop.5:[CNode]6:1bprop.5:[CNode]6:1"REF::S-Prim-MakeTuple:2:Default/S-Prim-MakeTuple-op2bprop.5*
bprop.5:x*
bprop.5:out*
bprop.5:dout2
bprop.5:[CNode]6:1:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPb&
S-Prim-MakeTuple:2S-Prim-MakeTuple
bprop.3:doutbprop.3:[CNode]4:1bprop.3:[CNode]4:1"REF::S-Prim-MakeTuple:2:Default/S-Prim-MakeTuple-op0bprop.3*
bprop.3:x*
bprop.3:out*
bprop.3:dout2
bprop.3:[CNode]4:1:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7Pb&
S-Prim-MakeTuple:2S-Prim-MakeTupleh

View File

@ -1,14 +1,14 @@
0.1.1 MindSpore*1.7.0:¤

0.1.1 MindSpore*1.7.0:¢

bprop.39:xbprop.39:[CNode]40:1bprop.39:[CNode]40:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op19
z
bprop.39:[CNode]40:1bprop.39:[CNode]41:3bprop.39:[CNode]41:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op20bprop.39*
bprop.20:xbprop.20:[CNode]21:1bprop.20:[CNode]21:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op6
y
bprop.20:[CNode]21:1bprop.20:[CNode]22:3bprop.20:[CNode]22:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op7bprop.20*
bprop.39:x*
bprop.39:out*
bprop.39:dout2
bprop.39:[CNode]41:3:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
bprop.20:x*
bprop.20:out*
bprop.20:dout2
bprop.20:[CNode]22:3:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:¸
˜
bprop.198:xbprop.198:[CNode]199:1bprop.198:[CNode]199:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:/Default/S-Prim-hyper_map[zeros_like_leaf]-op112
<EFBFBD>
bprop.198:[CNode]199:1bprop.198:[CNode]200:3bprop.198:[CNode]200:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op113 bprop.198*
bprop.198:x*
bprop.198:out*
bprop.198:dout2
bprop.198:[CNode]200:3:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
0.1.1 MindSpore*1.7.0:

bprop.137:xbprop.137:[CNode]138:1bprop.137:[CNode]138:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op67

bprop.137:[CNode]138:1bprop.137:[CNode]139:3bprop.137:[CNode]139:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op68 bprop.137*
bprop.137:x*
bprop.137:out*
bprop.137:dout2
bprop.137:[CNode]139:3:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:¸
˜
bprop.194:xbprop.194:[CNode]195:1bprop.194:[CNode]195:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:/Default/S-Prim-hyper_map[zeros_like_leaf]-op110
<EFBFBD>
bprop.194:[CNode]195:1bprop.194:[CNode]196:3bprop.194:[CNode]196:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op111 bprop.194*
bprop.194:x*
bprop.194:out*
bprop.194:dout2
bprop.194:[CNode]196:3:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
0.1.1 MindSpore*1.7.0:

bprop.133:xbprop.133:[CNode]134:1bprop.133:[CNode]134:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op65

bprop.133:[CNode]134:1bprop.133:[CNode]135:3bprop.133:[CNode]135:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op66 bprop.133*
bprop.133:x*
bprop.133:out*
bprop.133:dout2
bprop.133:[CNode]135:3:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -1,16 +1,20 @@
0.1.1 MindSpore*1.7.0:÷

bprop.106:xbprop.106:[CNode]107:1bprop.106:[CNode]107:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op57

bprop.106:ybprop.106:[CNode]108:3bprop.106:[CNode]108:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op58
˜
bprop.106:[CNode]107:1
bprop.106:[CNode]108:3bprop.106:[CNode]109:4bprop.106:[CNode]109:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op59 bprop.106*
bprop.106:x*
bprop.106:y*
bprop.106:out*
bprop.106:dout2
bprop.106:[CNode]109:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
0.1.1 MindSpore*1.7.0:Þ

bprop.77:xbprop.77:[CNode]78:1bprop.77:[CNode]78:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op36

bprop.77:ybprop.77:[CNode]79:3bprop.77:[CNode]79:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op37
<EFBFBD>
bprop.77:[CNode]78:1
bprop.77:[CNode]79:3bprop.77:[CNode]80:4bprop.77:[CNode]80:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op38bprop.77*
bprop.77:x*
bprop.77:y*
bprop.77:out*
bprop.77:dout2
bprop.77:[CNode]80:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -1,16 +1,20 @@
0.1.1 MindSpore*1.7.0:÷

bprop.101:xbprop.101:[CNode]102:1bprop.101:[CNode]102:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op54

bprop.101:ybprop.101:[CNode]103:3bprop.101:[CNode]103:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op55
˜
bprop.101:[CNode]102:1
bprop.101:[CNode]103:3bprop.101:[CNode]104:4bprop.101:[CNode]104:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op56 bprop.101*
bprop.101:x*
bprop.101:y*
bprop.101:out*
bprop.101:dout2
bprop.101:[CNode]104:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
0.1.1 MindSpore*1.7.0:Þ

bprop.72:xbprop.72:[CNode]73:1bprop.72:[CNode]73:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op33

bprop.72:ybprop.72:[CNode]74:3bprop.72:[CNode]74:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op34
<EFBFBD>
bprop.72:[CNode]73:1
bprop.72:[CNode]74:3bprop.72:[CNode]75:4bprop.72:[CNode]75:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op35bprop.72*
bprop.72:x*
bprop.72:y*
bprop.72:out*
bprop.72:dout2
bprop.72:[CNode]75:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -1,20 +1,20 @@
0.1.1 MindSpore*1.7.0:©

bprop.60:startbprop.60:[CNode]61:1bprop.60:[CNode]61:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op30
bprop.41:startbprop.41:[CNode]42:1bprop.41:[CNode]42:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op17

bprop.60:stopbprop.60:[CNode]62:3bprop.60:[CNode]62:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op31
bprop.41:stopbprop.41:[CNode]43:3bprop.41:[CNode]43:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op18

bprop.60:numbprop.60:[CNode]63:4bprop.60:[CNode]63:4"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op32
bprop.41:numbprop.41:[CNode]44:4bprop.41:[CNode]44:4"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op19
¦
bprop.60:[CNode]61:1
bprop.60:[CNode]62:3
bprop.60:[CNode]63:4bprop.60:[CNode]64:5bprop.60:[CNode]64:5"REF::S-Prim-MakeTuple:6:Default/S-Prim-MakeTuple-op33bprop.60*
bprop.60:start*
bprop.60:stop*
bprop.60:num*
bprop.60:out*
bprop.60:dout2
bprop.60:[CNode]64:5:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:6S-Prim-MakeTuple
bprop.41:[CNode]42:1
bprop.41:[CNode]43:3
bprop.41:[CNode]44:4bprop.41:[CNode]45:5bprop.41:[CNode]45:5"REF::S-Prim-MakeTuple:6:Default/S-Prim-MakeTuple-op20bprop.41*
bprop.41:start*
bprop.41:stop*
bprop.41:num*
bprop.41:out*
bprop.41:dout2
bprop.41:[CNode]45:5:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:6S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,16 +1,20 @@
0.1.1 MindSpore*1.7.0:÷

bprop.111:xbprop.111:[CNode]112:1bprop.111:[CNode]112:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op60

bprop.111:ybprop.111:[CNode]113:3bprop.111:[CNode]113:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op61
˜
bprop.111:[CNode]112:1
bprop.111:[CNode]113:3bprop.111:[CNode]114:4bprop.111:[CNode]114:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op62 bprop.111*
bprop.111:x*
bprop.111:y*
bprop.111:out*
bprop.111:dout2
bprop.111:[CNode]114:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
0.1.1 MindSpore*1.7.0:Þ

bprop.82:xbprop.82:[CNode]83:1bprop.82:[CNode]83:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op39

bprop.82:ybprop.82:[CNode]84:3bprop.82:[CNode]84:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op40
<EFBFBD>
bprop.82:[CNode]83:1
bprop.82:[CNode]84:3bprop.82:[CNode]85:4bprop.82:[CNode]85:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op41bprop.82*
bprop.82:x*
bprop.82:y*
bprop.82:out*
bprop.82:dout2
bprop.82:[CNode]85:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -2,13 +2,13 @@
0.1.1 MindSpore*1.7.0:¤

bprop.52:xbprop.52:[CNode]53:1bprop.52:[CNode]53:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op26
bprop.33:xbprop.33:[CNode]34:1bprop.33:[CNode]34:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op13
z
bprop.52:[CNode]53:1bprop.52:[CNode]54:3bprop.52:[CNode]54:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op27bprop.52*
bprop.33:[CNode]34:1bprop.33:[CNode]35:3bprop.33:[CNode]35:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op14bprop.33*
bprop.52:x*
bprop.52:out*
bprop.52:dout2
bprop.52:[CNode]54:3:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
bprop.33:x*
bprop.33:out*
bprop.33:dout2
bprop.33:[CNode]35:3:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -1,16 +1,20 @@
0.1.1 MindSpore*1.7.0:÷

bprop.116:xbprop.116:[CNode]117:1bprop.116:[CNode]117:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op63

bprop.116:ybprop.116:[CNode]118:3bprop.116:[CNode]118:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op64
˜
bprop.116:[CNode]117:1
bprop.116:[CNode]118:3bprop.116:[CNode]119:4bprop.116:[CNode]119:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op65 bprop.116*
bprop.116:x*
bprop.116:y*
bprop.116:out*
bprop.116:dout2
bprop.116:[CNode]119:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
0.1.1 MindSpore*1.7.0:Þ

bprop.87:xbprop.87:[CNode]88:1bprop.87:[CNode]88:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op42

bprop.87:ybprop.87:[CNode]89:3bprop.87:[CNode]89:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op43
<EFBFBD>
bprop.87:[CNode]88:1
bprop.87:[CNode]89:3bprop.87:[CNode]90:4bprop.87:[CNode]90:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op44bprop.87*
bprop.87:x*
bprop.87:y*
bprop.87:out*
bprop.87:dout2
bprop.87:[CNode]90:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -2,19 +2,19 @@
0.1.1 MindSpore*1.7.0:Þ

bprop.86:xbprop.86:[CNode]87:1bprop.86:[CNode]87:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op45
bprop.57:xbprop.57:[CNode]58:1bprop.57:[CNode]58:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op24

bprop.86:ybprop.86:[CNode]88:3bprop.86:[CNode]88:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op46
bprop.57:ybprop.57:[CNode]59:3bprop.57:[CNode]59:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op25
<EFBFBD>
bprop.86:[CNode]87:1
bprop.86:[CNode]88:3bprop.86:[CNode]89:4bprop.86:[CNode]89:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op47bprop.86*
bprop.57:[CNode]58:1
bprop.57:[CNode]59:3bprop.57:[CNode]60:4bprop.57:[CNode]60:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op26bprop.57*
bprop.86:x*
bprop.57:x*
bprop.86:y*
bprop.86:out*
bprop.86:dout2
bprop.86:[CNode]89:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
bprop.57:y*
bprop.57:out*
bprop.57:dout2
bprop.57:[CNode]60:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -2,13 +2,13 @@
0.1.1 MindSpore*1.7.0:¢

bprop.12:xbprop.12:[CNode]13:1bprop.12:[CNode]13:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op5
bprop.10:xbprop.10:[CNode]11:1bprop.10:[CNode]11:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op3
y
bprop.12:[CNode]13:1bprop.12:[CNode]14:3bprop.12:[CNode]14:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op6bprop.12*
bprop.10:[CNode]11:1bprop.10:[CNode]12:3bprop.10:[CNode]12:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op4bprop.10*
bprop.12:x*
bprop.12:out*
bprop.12:dout2
bprop.12:[CNode]14:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPb&
bprop.10:x*
bprop.10:out*
bprop.10:dout2
bprop.10:[CNode]12:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7Pb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:
0.1.1 MindSpore*1.7.0:
<EFBFBD>
bprop.8:xbprop.8:[CNode]9:1bprop.8:[CNode]9:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op3
u
bprop.8:[CNode]9:1bprop.8:[CNode]10:3bprop.8:[CNode]10:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op4bprop.8*
bprop.8:x*
bprop.8:out*
bprop.8:dout2
bprop.8:[CNode]10:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
bprop.6:xbprop.6:[CNode]7:1bprop.6:[CNode]7:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op1
s
bprop.6:[CNode]7:1bprop.6:[CNode]8:3bprop.6:[CNode]8:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op2bprop.6*
bprop.6:x*
bprop.6:out*
bprop.6:dout2
bprop.6:[CNode]8:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7Pb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:¶

bprop.149:xbprop.149:[CNode]150:1bprop.149:[CNode]150:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op82
bprop.132:xbprop.132:[CNode]133:1bprop.132:[CNode]133:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op68

bprop.149:[CNode]150:1bprop.149:[CNode]151:3bprop.149:[CNode]151:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op83 bprop.149*
bprop.149:x*
bprop.149:out*
bprop.149:dout2
bprop.149:[CNode]151:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
bprop.132:[CNode]133:1bprop.132:[CNode]134:3bprop.132:[CNode]134:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op69 bprop.132*
bprop.132:x*
bprop.132:out*
bprop.132:dout2
bprop.132:[CNode]134:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7Pb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,16 +1,18 @@
0.1.1 MindSpore*1.7.0:ý

bprop.121:xbprop.121:[CNode]122:1bprop.121:[CNode]122:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op66
š
bprop.121:axisbprop.121:[CNode]123:3bprop.121:[CNode]123:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op67
˜
bprop.121:[CNode]122:1
bprop.121:[CNode]123:3bprop.121:[CNode]124:4bprop.121:[CNode]124:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op68 bprop.121*
bprop.121:x*
bprop.121:axis*
bprop.121:out*
bprop.121:dout2
bprop.121:[CNode]124:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
0.1.1 MindSpore*1.7.0:ä

bprop.92:xbprop.92:[CNode]93:1bprop.92:[CNode]93:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op45

bprop.92:axisbprop.92:[CNode]94:3bprop.92:[CNode]94:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op46
<EFBFBD>
bprop.92:[CNode]93:1
bprop.92:[CNode]94:3bprop.92:[CNode]95:4bprop.92:[CNode]95:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op47bprop.92*
bprop.92:x*
bprop.92:axis*
bprop.92:out*
bprop.92:dout2
bprop.92:[CNode]95:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,16 +1,18 @@
0.1.1 MindSpore*1.7.0:ý

bprop.126:xbprop.126:[CNode]127:1bprop.126:[CNode]127:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op69
š
bprop.126:axisbprop.126:[CNode]128:3bprop.126:[CNode]128:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op70
˜
bprop.126:[CNode]127:1
bprop.126:[CNode]128:3bprop.126:[CNode]129:4bprop.126:[CNode]129:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op71 bprop.126*
bprop.126:x*
bprop.126:axis*
bprop.126:out*
bprop.126:dout2
bprop.126:[CNode]129:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
0.1.1 MindSpore*1.7.0:ç

bprop.97:xbprop.97:[CNode]98:1bprop.97:[CNode]98:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op48

bprop.97:axisbprop.97:[CNode]99:3bprop.97:[CNode]99:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op49

bprop.97:[CNode]98:1
bprop.97:[CNode]99:3bprop.97:[CNode]100:4bprop.97:[CNode]100:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op50bprop.97*
bprop.97:x*
bprop.97:axis*
bprop.97:out*
bprop.97:dout2
bprop.97:[CNode]100:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:5S-Prim-MakeTuple
S-Prim-MakeTuple:5S-Prim-MakeTupleh

View File

@ -2,13 +2,13 @@
0.1.1 MindSpore*1.7.0:¤

bprop.56:xbprop.56:[CNode]57:1bprop.56:[CNode]57:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op28
bprop.37:xbprop.37:[CNode]38:1bprop.37:[CNode]38:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op15
z
bprop.56:[CNode]57:1bprop.56:[CNode]58:3bprop.56:[CNode]58:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op29bprop.56*
bprop.37:[CNode]38:1bprop.37:[CNode]39:3bprop.37:[CNode]39:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op16bprop.37*
bprop.56:x*
bprop.56:out*
bprop.56:dout2
bprop.56:[CNode]58:3:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
bprop.37:x*
bprop.37:out*
bprop.37:dout2
bprop.37:[CNode]39:3:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -1,31 +1,31 @@
0.1.1 MindSpore*1.7.0:ø
š
bprop.153:condbprop.153:[CNode]154:1bprop.153:[CNode]154:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op84
bprop.136:condbprop.136:[CNode]137:1bprop.136:[CNode]137:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op70

bprop.153:xbprop.153:[CNode]155:3bprop.153:[CNode]155:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op85
bprop.136:xbprop.136:[CNode]138:3bprop.136:[CNode]138:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op71
š
bprop.153:cond
bprop.153:dout
bprop.153:[CNode]155:3bprop.153:[CNode]156:4bprop.153:[CNode]156:4"REF::S-Prim-Select:5:Default/S-Prim-Select-op86
bprop.136:cond
bprop.136:dout
bprop.136:[CNode]138:3bprop.136:[CNode]139:4bprop.136:[CNode]139:4"REF::S-Prim-Select:5:Default/S-Prim-Select-op72

bprop.153:ybprop.153:[CNode]157:6bprop.153:[CNode]157:6"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op87
bprop.136:ybprop.136:[CNode]140:6bprop.136:[CNode]140:6"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op73
š
bprop.153:cond
bprop.153:[CNode]157:6
bprop.153:doutbprop.153:[CNode]158:7bprop.153:[CNode]158:7"REF::S-Prim-Select:5:Default/S-Prim-Select-op88
bprop.136:cond
bprop.136:[CNode]140:6
bprop.136:doutbprop.136:[CNode]141:7bprop.136:[CNode]141:7"REF::S-Prim-Select:5:Default/S-Prim-Select-op74
°
bprop.153:[CNode]154:1
bprop.153:[CNode]156:4
bprop.153:[CNode]158:7bprop.153:[CNode]159:8bprop.153:[CNode]159:8"REF::S-Prim-MakeTuple:9:Default/S-Prim-MakeTuple-op89 bprop.153*
bprop.153:cond*
bprop.153:x*
bprop.153:y*
bprop.153:out*
bprop.153:dout2
bprop.153:[CNode]159:8:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPb&
S-Prim-MakeTuple:9S-Prim-MakeTuplebv
bprop.136:[CNode]137:1
bprop.136:[CNode]139:4
bprop.136:[CNode]141:7bprop.136:[CNode]142:8bprop.136:[CNode]142:8"REF::S-Prim-MakeTuple:9:Default/S-Prim-MakeTuple-op75 bprop.136*
bprop.136:cond*
bprop.136:x*
bprop.136:y*
bprop.136:out*
bprop.136:dout2
bprop.136:[CNode]142:8:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7PbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]bv
S-Prim-Select:5 S-Prim-Select
output_names€Š Zoutput€3
input_names€ŠZ condition€ŠZx€ŠZy€bH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
input_names€ŠZ condition€ŠZx€ŠZy€b&
S-Prim-MakeTuple:9S-Prim-MakeTupleh

View File

@ -1,12 +1,12 @@
0.1.1 MindSpore*1.7.0:¶

bprop.141:xbprop.141:[CNode]142:1bprop.141:[CNode]142:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op78
bprop.124:xbprop.124:[CNode]125:1bprop.124:[CNode]125:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op64

bprop.141:[CNode]142:1bprop.141:[CNode]143:3bprop.141:[CNode]143:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op79 bprop.141*
bprop.141:x*
bprop.141:out*
bprop.141:dout2
bprop.141:[CNode]143:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPbH
bprop.124:[CNode]125:1bprop.124:[CNode]126:3bprop.124:[CNode]126:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op65 bprop.124*
bprop.124:x*
bprop.124:out*
bprop.124:dout2
bprop.124:[CNode]126:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7PbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -2,13 +2,13 @@
0.1.1 MindSpore*1.7.0:¤

bprop.48:xbprop.48:[CNode]49:1bprop.48:[CNode]49:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op24
bprop.29:xbprop.29:[CNode]30:1bprop.29:[CNode]30:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op11
z
bprop.48:[CNode]49:1bprop.48:[CNode]50:3bprop.48:[CNode]50:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op25bprop.48*
bprop.29:[CNode]30:1bprop.29:[CNode]31:3bprop.29:[CNode]31:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op12bprop.29*
bprop.48:x*
bprop.48:out*
bprop.48:dout2
bprop.48:[CNode]50:3:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
S-Prim-MakeTuple:4S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
bprop.29:x*
bprop.29:out*
bprop.29:dout2
bprop.29:[CNode]31:3:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -1,16 +1,16 @@
0.1.1 MindSpore*1.7.0:ù
0.1.1 MindSpore*1.7.0:÷

bprop.177:xbprop.177:[CNode]178:1bprop.177:[CNode]178:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op99
bprop.116:xbprop.116:[CNode]117:1bprop.116:[CNode]117:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op54

bprop.116:ybprop.116:[CNode]118:3bprop.116:[CNode]118:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:.Default/S-Prim-hyper_map[zeros_like_leaf]-op55
˜
bprop.177:ybprop.177:[CNode]179:3bprop.177:[CNode]179:3"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:/Default/S-Prim-hyper_map[zeros_like_leaf]-op100

bprop.177:[CNode]178:1
bprop.177:[CNode]179:3bprop.177:[CNode]180:4bprop.177:[CNode]180:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op101 bprop.177*
bprop.177:x*
bprop.177:y*
bprop.177:out*
bprop.177:dout2
bprop.177:[CNode]180:4:@33ebc53e3c87fc0fc2a3c9506915190e5926db400bd5e66a746ceb46c40b05fdPb&
bprop.116:[CNode]117:1
bprop.116:[CNode]118:3bprop.116:[CNode]119:4bprop.116:[CNode]119:4"REF::S-Prim-MakeTuple:5:Default/S-Prim-MakeTuple-op56 bprop.116*
bprop.116:x*
bprop.116:y*
bprop.116:out*
bprop.116:dout2
bprop.116:[CNode]119:4:@c5aceb168aeeb852fdb52d5efe3a7782f6e6e7eed35f19d4ebe07d940164ec2fPb&
S-Prim-MakeTuple:5S-Prim-MakeTuplebH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]h

View File

@ -2,13 +2,13 @@
0.1.1 MindSpore*1.7.0:¢

bprop.16:xbprop.16:[CNode]17:1bprop.16:[CNode]17:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op7
bprop.14:xbprop.14:[CNode]15:1bprop.14:[CNode]15:1"(REF::S-Prim-hyper_map[zeros_like_leaf]:2:-Default/S-Prim-hyper_map[zeros_like_leaf]-op5
y
bprop.16:[CNode]17:1bprop.16:[CNode]18:3bprop.16:[CNode]18:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op8bprop.16*
bprop.14:[CNode]15:1bprop.14:[CNode]16:3bprop.14:[CNode]16:3"REF::S-Prim-MakeTuple:4:Default/S-Prim-MakeTuple-op6bprop.14*
bprop.16:x*
bprop.16:out*
bprop.16:dout2
bprop.16:[CNode]18:3:@3e6b4b1515849c402b76774467f7d4737b4a5c7f70c42f3deb212cc538028f2bPbH
bprop.14:x*
bprop.14:out*
bprop.14:dout2
bprop.14:[CNode]16:3:@ceb6bc8e8d5c40f04ae14aae46495a62e23d11908cae60a0b9eecf0ca9718ac7PbH
#S-Prim-hyper_map[zeros_like_leaf]:2!S-Prim-hyper_map[zeros_like_leaf]b&
S-Prim-MakeTuple:4S-Prim-MakeTuple
S-Prim-MakeTuple:4S-Prim-MakeTupleh

View File

@ -521,7 +521,7 @@ class Reshape(PrimitiveWithInfer):
if len(min_shape) != shape_rank or len(max_shape) != shape_rank:
raise RuntimeError("The primitive[Reshape]'s input[shape] min or max value not math the shape rank.")
for i in range(shape_rank):
if min_shape[i] == max_shape[i] and min_shape[i] != 1:
if min_shape[i] == max_shape[i]:
out_shape[i] = min_shape[i]
elif is_shape_unknown(x_shp) and "max_shape" in x:
# when dynamic memory allocation is supported, max_shape can be left out
@ -529,9 +529,57 @@ class Reshape(PrimitiveWithInfer):
max_shape = [int(np.prod(x["max_shape"]))] * shape_rank
return out_shape, min_shape, max_shape
def _update_shape_range(self, out, x, shape_v, neg_index, dim_prod):
""" update min and max shape of output when input shape is dynamic"""
x_min_shape = x['shape']
x_max_shape = x['shape']
if 'max_shape' in x:
x_max_shape = x['max_shape']
if 'min_shape' in x:
x_min_shape = x['min_shape']
max_arr_prod = np.prod(x_max_shape)
min_arr_prod = np.prod(x_min_shape)
max_shape = list(shape_v)
min_shape = list(shape_v)
if neg_index != -1:
max_shape[neg_index] = int(max_arr_prod / dim_prod)
min_shape[neg_index] = int(min_arr_prod / dim_prod)
out['max_shape'] = tuple(max_shape)
out['min_shape'] = tuple(min_shape)
return out
def _update_shape_and_value(self, out, x, shape_v, dim_prod, neg_index):
""" update shape, value and min / max value of output when input shape is known"""
x_shp = x['shape']
if dim_prod <= 0:
raise ValueError(f"For '{self.name}', the shape of 'input_x' is {x_shp}, "
f"the value of 'input_shape' is {shape_v}. "
f"The product of 'input_shape' should > 0, but got {dim_prod}.")
arr_prod = np.prod(x_shp)
if neg_index != -1:
shape_v[neg_index] = int(arr_prod / dim_prod)
dim_prod *= shape_v[neg_index]
if dim_prod != arr_prod:
raise ValueError(f"For '{self.name}', the shape of 'input_x' is {x_shp}, "
f"the value of 'input_shape' value is {shape_v}. "
f"The product of the shape of 'input_x' should be equal to that of 'input_shape', "
f"but got {arr_prod} and {dim_prod}.")
out['shape'] = tuple(shape_v)
if x['value'] is not None:
out['value'] = Tensor(x['value'].asnumpy().reshape(shape_v))
if ('min_value' in x and 'max_value' in x):
ret_min_value = np.array(x['min_value']).reshape(shape_v)
ret_max_value = np.array(x['max_value']).reshape(shape_v)
ret_min_value = tuple(ret_min_value.tolist())
ret_max_value = tuple(ret_max_value.tolist())
out['min_value'] = ret_min_value
out['max_value'] = ret_max_value
return out
def __infer__(self, x, shape):
shape_v = shape['value']
x_shp = x['shape']
validator.check_subclass("x", x['dtype'], mstype.tensor, self.name)
# for shape is not constant
if shape_v is None:
@ -569,50 +617,14 @@ class Reshape(PrimitiveWithInfer):
else:
dim_prod *= shp_i
if is_shape_unknown(x_shp):
if 'max_shape' in x:
x_max_shape = x['max_shape']
else:
x_max_shape = x['shape']
if 'min_shape' in x:
x_min_shape = x['min_shape']
else:
x_min_shape = x['shape']
max_arr_prod = np.prod(x_max_shape)
min_arr_prod = np.prod(x_min_shape)
max_shape = list(shape_v)
min_shape = list(shape_v)
if neg_index != -1:
max_shape[neg_index] = int(max_arr_prod / dim_prod)
min_shape[neg_index] = int(min_arr_prod / dim_prod)
out = {'shape': shape_v,
'dtype': x['dtype'],
'value': None}
out = {'shape': shape_v,
'dtype': x['dtype'],
'value': None,
'max_shape': tuple(max_shape),
'min_shape': tuple(min_shape)}
if is_shape_unknown(x['shape']):
out = self._update_shape_range(out, x, shape_v, neg_index, dim_prod)
else:
arr_prod = np.prod(x_shp)
if dim_prod <= 0:
raise ValueError(f"For '{self.name}', the shape of 'input_x' is {x_shp}, "
f"the value of 'input_shape' is {shape_v}. "
f"The product of 'input_shape' should > 0, but got {dim_prod}.")
if neg_index != -1:
shape_v[neg_index] = int(arr_prod / dim_prod)
dim_prod *= shape_v[neg_index]
if dim_prod != arr_prod:
raise ValueError(f"For '{self.name}', the shape of 'input_x' is {x_shp}, "
f"the value of 'input_shape' value is {shape_v}. "
f"The product of the shape of 'input_x' should be equal to product of 'input_shape', "
f"but product of the shape of 'input_x' is {arr_prod}, "
f"product of 'input_shape' is {dim_prod}.")
value = None
if x['value'] is not None:
value = Tensor(x['value'].asnumpy().reshape(shape_v))
out = {'shape': tuple(shape_v),
'dtype': x['dtype'],
'value': value}
out = self._update_shape_and_value(out, x, shape_v, dim_prod, neg_index)
return out
@ -1352,20 +1364,32 @@ class Fill(PrimitiveWithInfer):
def __infer__(self, dtype, dims, x):
validator.check_value_type("shape", dims['value'], [tuple], self.name)
validator.check_value_type("value", x['value'], [numbers.Number, bool], self.name)
for i, item in enumerate(dims['value']):
validator.check_positive_int(item, f'dims[{i}]', self.name)
valid_dtypes = [mstype.bool_, mstype.int8, mstype.int16, mstype.int32, mstype.int64,
mstype.uint8, mstype.uint16, mstype.uint32, mstype.uint64,
mstype.float16, mstype.float32, mstype.float64, mstype.complex64,
mstype.complex128]
validator.check_types_same_and_valid({"value": dtype['value']}, valid_dtypes, self.name)
x_nptype = mstype.dtype_to_nptype(dtype['value'])
ret = np.full(dims['value'], x['value'], x_nptype)
out = {
'value': Tensor(ret),
'shape': dims['value'],
'dtype': x['dtype'],
}
if -1 not in dims['value']:
for i, item in enumerate(dims['value']):
validator.check_positive_int(item, f'dims[{i}]', self.name)
ret = np.full(dims['value'], x['value'], x_nptype)
out = {
'value': Tensor(ret),
'shape': dims['value'],
'dtype': x['dtype'],
}
else:
out = {
'value': None,
'shape': dims['value'],
'dtype': x['dtype'],
}
if ('min_value' in dims and 'max_value' in dims):
min_ret_shape = dims['min_shape']
max_ret_shape = dims['max_shape']
out['min_shape'] = min_ret_shape
out['max_shape'] = max_ret_shape
return out
@ -2800,6 +2824,22 @@ class Stack(PrimitiveWithInfer):
out = {'shape': all_shape,
'dtype': x_type[0],
'value': infered_value}
if ('min_value' in value and 'max_value' in value):
min_value_array = []
max_value_array = []
infered_min_value = None
infered_max_value = None
for i in range(len(value['min_value'])):
cur_min_value = value['min_value'][i]
cur_max_value = value['max_value'][i]
min_value_array.append(np.array(cur_min_value))
max_value_array.append(np.array(cur_max_value))
infered_min_value = np.stack(min_value_array, axis=self.axis)
infered_max_value = np.stack(max_value_array, axis=self.axis)
infered_min_value = tuple(infered_min_value.tolist())
infered_max_value = tuple(infered_max_value.tolist())
out['min_value'] = infered_min_value
out['max_value'] = infered_max_value
return out
@ -4170,8 +4210,8 @@ class TensorScatterUpdate(PrimitiveWithInfer):
def __init__(self):
self.init_prim_io_names(inputs=['input_x', 'indices', 'updates'], outputs=['y'])
def _infer_min_max_value(self, input_x_value, indices_value, updates_value):
"""TensorScatterUpdate infer min max value"""
def _infer_specified_value(self, input_x_value, indices_value, updates_value):
"""Calculate min/max value for output of TensorScatterUpdate op"""
if isinstance(input_x_value, tuple):
input_x_value = list(input_x_value)
if isinstance(input_x_value, (Tensor, Tensor_)):
@ -4184,13 +4224,11 @@ class TensorScatterUpdate(PrimitiveWithInfer):
output = tuple(input_x.tolist())
return output
def infer_min_value(self, input_x_value, indices_value, updates_value):
"""TensorScatterUpdate infer min value"""
return self._infer_min_max_value(input_x_value, indices_value, updates_value)
def _infer_min_value(self, input_x_value, indices_value, updates_value):
return self._infer_specified_value(input_x_value, indices_value, updates_value)
def infer_max_value(self, input_x_value, indices_value, updates_value):
"""TensorScatterUpdate infer max value"""
return self._infer_min_max_value(input_x_value, indices_value, updates_value)
def _infer_max_value(self, input_x_value, indices_value, updates_value):
return self._infer_specified_value(input_x_value, indices_value, updates_value)
def infer_shape(self, input_x_shape, indices_shape, updates_shape):
if len(indices_shape) < 2:

View File

@ -235,8 +235,8 @@ class Add(_MathBinaryOp):
>>> print(output.dtype)
Float32
"""
def _add_infer_special_value(self, a, b):
"""Add infer min max value"""
def _infer_specified_add_value(self, a, b):
"""Calculate min/max value for output for Add op"""
if a is not None and b is not None:
if isinstance(a, (Tensor, Tensor_)):
a = a.asnumpy()
@ -249,13 +249,13 @@ class Add(_MathBinaryOp):
return out
return None
def infer_min_value(self, x, y):
"""Add infer min value"""
return _add_infer_special_value(x, y)
def _infer_min_value(self, x, y):
"""Calculate min value for output for Add op"""
return self._infer_specified_add_value(x, y)
def infer_max_value(self, x, y):
"""Add infer min value"""
return _add_infer_special_value(x, y)
def _infer_max_value(self, x, y):
"""Calculate max value for output for Add op"""
return self._infer_specified_add_value(x, y)
def infer_value(self, x, y):
if x is not None and y is not None:
@ -1937,8 +1937,8 @@ class Mul(_MathBinaryOp):
>>> print(output)
[ 4. 10. 18.]
"""
def _infer_min_max_value(self, x, y):
"""Mul infer min max value"""
def _infer_specified_mul_value(self, x, y):
"""Calculate min/max value for output of Mul op"""
if x is not None and y is not None:
if isinstance(x, (Tensor, Tensor_)):
x = x.asnumpy()
@ -1951,13 +1951,13 @@ class Mul(_MathBinaryOp):
return out
return None
def infer_min_value(self, x, y):
"""Mul infer min value"""
return self._infer_min_max_value(x, y)
def _infer_min_value(self, x, y):
"""Calculate min value for output for Mul op"""
return self._infer_specified_mul_value(x, y)
def infer_max_value(self, x, y):
"""Mul infer max value"""
return self._infer_min_max_value(x, y)
def _infer_max_value(self, x, y):
"""Calculate max value for output for Mul op"""
return self._infer_specified_mul_value(x, y)
def infer_dtype(self, x_dtype, y_dtype):
mul_valid_type = mstype.number_type + (mstype.bool_,)
@ -2872,8 +2872,9 @@ class Div(_MathBinaryOp):
>>> print(output.dtype)
Float32
"""
def _div_infer_special_value(self, x, y):
"""Div infer min max value"""
def _infer_specified_div_value(self, x, y):
"""Calculate min/max value for output of Div op"""
if x is not None and y is not None:
if isinstance(x, (Tensor, Tensor_)):
x = x.asnumpy()
@ -2886,13 +2887,13 @@ class Div(_MathBinaryOp):
return out
return None
def infer_min_value(self, x, y):
"""Div infer min value"""
return _div_infer_special_value(x, y)
def _infer_min_value(self, x, y):
"""Calculate min value for output for Div op"""
return self._infer_specified_div_value(x, y)
def infer_max_value(self, x, y):
"""Div infer max value"""
return _div_infer_special_value(x, y)
def _infer_max_value(self, x, y):
"""Calculate max value for output for Div op"""
return self._infer_specified_div_value(x, y)
def infer_value(self, x, y):
if x is not None and y is not None:

View File

@ -599,11 +599,11 @@ class PrimitiveWithInfer(Primitive):
has_min_value, min_values = get_specified_value(args, 'min_value')
has_max_value, max_values = get_specified_value(args, 'max_value')
if has_min_value and has_max_value:
if hasattr(self, 'infer_min_value'):
fn_infer_min_value = getattr(self, 'infer_min_value')
if hasattr(self, '_infer_min_value'):
fn_infer_min_value = getattr(self, '_infer_min_value')
out['min_value'] = fn_infer_min_value(*min_values)
if hasattr(self, 'infer_max_value'):
fn_infer_max_value = getattr(self, 'infer_max_value')
if hasattr(self, '_infer_max_value'):
fn_infer_max_value = getattr(self, '_infer_max_value')
out['max_value'] = fn_infer_max_value(*max_values)
if not has_dynamic_shape(out['shape']):
return out