Contents

struct

SushiBLAS::Reduce::AxisSpan

Holds where one segmented reduction's inputs and outputs live, in element counts.

Declared in
include/SushiBLAS/support/reduce_axis.hpp

See also

include/SushiBLAS/support/README.md

Public attributes

std::size_t segments = 0

How many independent folds; equals the result's length.

std::size_t extent = 0

How many elements are folded into each segment's result.

std::size_t segment_stride = 0

Input elements between the starts of consecutive segments.

std::size_t element_stride = 1

Input elements between consecutive elements of one segment.

std::size_t out_stride = 1

Output elements between consecutive results.

std::size_t segments2 = 1

Holds the number of groups of segments; 1 when there is one segment axis.

See also

include/SushiBLAS/support/README.md

std::size_t segment_stride2 = 0

Input elements between the starts of consecutive segment groups.

Public member functions

std::size_t total_segments() const noexcept

How many folds in total: segments * segments2.