def contiguous_size

stelliferous@0.0.2.0 · tensor_view.bend:97

import stelliferous@0.0.2.0/tensor_view.bend as TensorView
import Base

97# Validate strides from the innermost axis outward. This avoids allocating a
98# second canonical View just to compare its metadata. Empty views bypass it.
99def contiguous_size(axes: List<&2,Axis>) -> Maybe<&2,U32>:
100  match axes:
101    case []: Some{1}
102    case Axis{extent,stride} <> tail: contiguous_axis(extent,stride,contiguous_size(tail))

Uses (5)

Used by (1)