Home
last modified time | relevance | path

Searched refs:_tuple_slice (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ops/composite/multitype_ops/
Dgetitem_impl.py49 _tuple_slice = _TupleSlice('tuple_slice') variable
103 return _tuple_slice(data, slice_index)
/third_party/mindspore/mindspore/numpy/
Darray_ops.py32 _tuple_slice, _expanded_shape, _seq_prod, _tuple_setitem, _iota, \
1793 indices = _broadcast_to(indices, _tuple_slice(shape_indices, None, axis),
1794 _tuple_slice(shape_arr, None, axis), ndim)
1795 indices = _broadcast_to(indices, _tuple_slice(shape_arr, None, axis + 1) +
1796 _tuple_slice(shape_indices, axis + 1, None), shape_arr, ndim)
2048 case_size_cond = _tuple_slice(shape_cond, 1, None)
2049 case_size_choice = _tuple_slice(shape_choice, 1, None)
2309 shape_stacked = (_tuple_slice(shape, None, axis) + _tuple_slice(shape, axis + 1, None) +
2310 _tuple_slice(F.shape(stacked_slices), 1, None))
Darray_creations.py36 _canonicalize_axis, _list_comprehensions, _ceil, _tuple_slice, _raise_unimplemented_error, \
469 …bounds_shape = _tuple_slice(bounds_shape, None, axis) + (1,) + _tuple_slice(bounds_shape, axis, No…
471 …iota_shape = _tuple_slice(iota_shape, None, axis) + (num,) + _tuple_slice(iota_shape, axis+1, None)
650 expanded_shape = _tuple_slice(shape, None, axis) + (1,) + _tuple_slice(shape, axis, None)
Dutils_const.py471 def _tuple_slice(tup, start, end): function
Dmath_ops.py44 _raise_unimplemented_error, _abs, _in, _tuple_slice, _check_is_inf
4175 arrays = F.reshape(arrays, (-1,) + _tuple_slice(F.shape(arrays), 2, None))