|
Poplar and PopLibs
|
This is the complete list of members for gcl::CollectiveBalancedReorder, including all inherited members.
| CollectiveBalancedReorder(poplar::Graph &graph_, poplar::Tensor tensor_, unsigned replicationFactor_, const poplar::DebugNameAndId &dnai_, bool allowElementMap=false, unsigned grainSize=1) | gcl::CollectiveBalancedReorder | |
| createCollectivesTensor(const poplar::Type &type, const std::string &debugPrefix) | gcl::CollectiveBalancedReorder | |
| createReplicaSlice(const poplar::Type &type) | gcl::CollectiveBalancedReorder | |
| getHostRearrangement() const | gcl::CollectiveBalancedReorder | inline |
| getReferenceShape() const | gcl::CollectiveBalancedReorder | inline |
| undoRearrangeForCollective(const poplar::Tensor &tensor) const | gcl::CollectiveBalancedReorder | |
| zeroPaddingInCollectiveTensor(poplar::Tensor &collectiveTensor, poplar::program::Sequence &prog) const | gcl::CollectiveBalancedReorder |