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 |