Searched refs:_add_unit_axes (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/numpy/ |
| D | utils.py | 22 from .utils_const import _tile_size, _add_unit_axes, _raise_type_error, _type_convert, \
80 shape = _add_unit_axes(F.shape(x), ndim, axis == -1)
|
| D | array_ops.py | 29 _check_same_type, _check_axis_valid, _add_unit_axes, _broadcast_tuples, \
1226 reps = _add_unit_axes(reps, ndim)
1228 shape = _add_unit_axes(shape, len(reps))
2177 …choices = F.reshape(choices, choices.shape[:1] + _add_unit_axes(choices.shape[1:], len(shape_choic…
|
| D | utils_const.py | 365 def _add_unit_axes(shape, ndim, append=False): function
|
| D | math_ops.py | 42 _check_dtype, _list_comprehensions, _tuple_setitem, _add_unit_axes, _seq_prod, \
2880 a = F.reshape(a, _add_unit_axes(shape_a, 2*ndim, True))
2881 a = F.tile(a, _add_unit_axes(shape_b, 2*ndim, False))
|