class

`SushiAI::Eager::ScopePartitioner`

Cuts a graph only where the caller opened or closed a scope.

Declared in: include/SushiAI/eager/scope_partitioner.hpp
Inherits: SushiAI::Eager::RegionPartitioner

Public member functions

virtual Partition assign(const Graph::Graph &graph, SushiRuntime::span< const Segment > segments, const Autograd::Provenance &provenance) const override

Gives segment k key k and its backward nodes key 2m - k, for m segments.

Parameters

segments: The forward trace's segments; they cover every forward node.
provenance: The record differentiate() filled for graph.