tvm_tirx::sym::iter_affine_map

Function subspace_divide

Source
pub fn subspace_divide(
    bindings: impl IntoIterator<Item = impl IntoExpr>,
    input_iters: &Map<Var, Range>,
    sub_iters: impl IntoIterator<Item = impl Into<Var>>,
    predicate: impl IntoExpr,
    check_level: IterMapLevel,
    analyzer: Option<&Analyzer>,
    simplify_trivial_iterators: bool,
) -> Result<Array<Array<IterMark>>>
Expand description

Divide each binding into an outer map of the iterators not in sub_iters and an inner map of those in it: one [outer, inner] pair of marks per binding, then a pair whose extents are the outer and inner predicates, or empty when the bindings cannot be divided (sym.SubspaceDivide).