Searched refs:target_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | layers.py | 116 def check_shape_equal(input_shape, param_name, func_name, target_shape): argument
125 if not isinstance(target_shape[0], list):
126 target_shape = [target_shape]
130 [len(item) for item in target_shape])
132 for item in target_shape:
179 def _check_shape_equal(input_shape, param_name, func_name, target_shape): argument
181 _LayerInputCheck.check_shape_equal(input_shape, param_name, func_name, target_shape)
|
| /third_party/mindspore/mindspore/numpy/ |
| D | array_ops.py | 1234 def _check_can_broadcast_to(shape, target_shape): argument
1237 ndim_target = len(target_shape)
1240 for i, j in zip(reversed(shape), reversed(target_shape)):
|