Function shape_abc

src/opr/test/misc.cpp:17–24

namespace {
void shape_abc(const TensorShape& shape, size_t axis, size_t& A, size_t& B, size_t& C) {
    auto acc_mul = [](const size_t* first, const size_t* last) {
        return std::accumulate(first, last, 1u, std::multiplies<size_t>());
    };
    A = acc_mul(shape.shape, shape.shape + axis);
    B = shape.shape[axis];
    C = acc_mul(shape.shape + axis + 1, shape.shape + shape.ndim);
}

void argsort_data_gen(HostTensorND& dest) {
    mgb_assert(dest.layout().ndim == 2 && dest.layout().is_contiguous());