Poplar and PopLibs
poplar::TensorRearranger Class Reference

TensorRearranger can be used to re-order the view on a tensor and to undo that re-ordering. More...

#include <TensorRearranger.hpp>

Public Member Functions

Tensor rearrange (const Tensor &t) const
 Rearrange a tensor.
 
Tensor undoRearrangement (const Tensor &t) const
 Undo the rearrangement done with the rearrange() method.
 
std::vector< Intervalrearrange (ArrayRef< Interval > is) const
 Rearrange intervals. More...
 
std::vector< IntervalundoRearrangement (ArrayRef< Interval > is) const
 Undo the rearrangement of intervals. More...
 

Detailed Description

TensorRearranger can be used to re-order the view on a tensor and to undo that re-ordering.

See also
Graph::getSimplifyingRearranger

Member Function Documentation

◆ rearrange()

std::vector< Interval > poplar::TensorRearranger::rearrange ( ArrayRef< Interval is) const

Rearrange intervals.

Parameters
isA list of intervals from the original tensor.
Returns
A list of equivalent intervals from the rearranged tensor.

◆ undoRearrangement()

std::vector< Interval > poplar::TensorRearranger::undoRearrangement ( ArrayRef< Interval is) const

Undo the rearrangement of intervals.

Parameters
isA list of intervals from the rearranged tensor.
Returns
A list of equivalent intervals from the original tensor.

The documentation for this class was generated from the following file: