Searched refs:ndim_to (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/numpy/ |
| D | utils.py | 84 def _broadcast_to(x, shape_cur, shape_to, ndim_to): argument
86 size = _tile_size(shape_cur, shape_to, ndim_to)
92 ndim_to = len(shape)
93 x = _expand(x, ndim_to)
94 return _broadcast_to(x, F.shape(x), shape, ndim_to)
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 730 def _broadcast_to(x, shape_cur, shape_to, ndim_to): argument
732 size = _tile_size(shape_cur, shape_to, ndim_to)
|