Contents

class

SushiAI::Eager::RegionPartitioner

Assigns every node of a differentiated graph a region key.

Declared in
include/SushiAI/eager/region_partitioner.hpp

See also

include/SushiAI/eager/README.md

Public member functions

virtual ~RegionPartitioner()=default

Destroys the partitioner.

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

Partitions graph.

Parameters

segments

The forward trace's segments; they cover every forward node.

provenance

The record differentiate() filled for graph.

Returns

A partition in which no node reads a value a higher key produces.