Searched refs:_broadcast_to (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/numpy/ |
| D | utils.py | 84 def _broadcast_to(x, shape_cur, shape_to, ndim_to): function
94 return _broadcast_to(x, F.shape(x), shape, ndim_to)
|
| D | array_ops.py | 26 _check_input_tensor, _broadcast_to, _to_tensor, _callable
1793 indices = _broadcast_to(indices, _tuple_slice(shape_indices, None, axis),
1795 indices = _broadcast_to(indices, _tuple_slice(shape_arr, None, axis + 1) +
|
| D | math_ops.py | 45 from .utils import _expand, _broadcast_to, _broadcast_to_shape, _check_input_tensor, \
1052 … sum_of_weights = _broadcast_to(sum_of_weights, sum_of_weights.shape, x_avg.shape, x_avg.ndim)
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 730 def _broadcast_to(x, shape_cur, shape_to, ndim_to): function
818 x1 = _broadcast_to(x1, shape1_aligned[:-2], shape_backbone, ndim_aligned)
819 x2 = _broadcast_to(x2, shape2_aligned[:-2], shape_backbone, ndim_aligned)
|