Searched refs:iter_shape (Results 1 – 4 of 4) sorted by relevance
| /dports/math/xtensor/xtensor-0.24.0/include/xtensor/ |
| H A D | xreducer.hpp | 292 …dynamic_shape<std::size_t> iter_shape = xtl::forward_sequence<dynamic_shape<std::size_t>, decltype… in reduce_immediate() local
372 iter_shape.erase(iter_shape.begin() + std::ptrdiff_t(last_ax), iter_shape.end()); in reduce_immediate()
381 iter_shape.erase(iter_shape.begin(), iter_shape.begin() + std::ptrdiff_t(last_ax + 1)); in reduce_immediate()
385 std::reverse(iter_shape.begin(), iter_shape.end()); in reduce_immediate()
393 xindex temp_idx(iter_shape.size()); in reduce_immediate()
394 auto next_idx = [&iter_shape, &iter_strides, &temp_idx]() { in reduce_immediate()
395 std::size_t i = iter_shape.size(); in reduce_immediate()
398 if (std::ptrdiff_t(temp_idx[i - 1]) >= std::ptrdiff_t(iter_shape[i - 1]) - 1) in reduce_immediate()
|
| /dports/math/py-numpy/numpy-1.20.3/numpy/random/ |
| H A D | _common.pxd | 80 cdef validate_output_shape(iter_shape, np.ndarray output) argument
|
| H A D | _common.pyx | 221 cdef validate_output_shape(iter_shape, np.ndarray output): argument
228 if iter_shape != output_shape:
|
| /dports/math/py-numpy/numpy-1.20.3/numpy/core/src/umath/ |
| H A D | ufunc_object.c | 2510 npy_intp iter_shape[NPY_MAXARGS]; in PyUFunc_GeneralizedFunction() local
2689 iter_shape[idim] = -1; in PyUFunc_GeneralizedFunction()
2736 iter_shape[j] = 1; in PyUFunc_GeneralizedFunction()
2751 iter_shape[j] = core_dim_sizes[ufunc->core_dim_ixs[core_index]]; in PyUFunc_GeneralizedFunction()
2764 printf(" %ld", iter_shape[j]); in PyUFunc_GeneralizedFunction()
2848 op_axes, iter_shape, 0); in PyUFunc_GeneralizedFunction()
|