Poplar and PopLibs
Sort.hpp File Reference

Functions for sorting tensors. More...

#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>
#include <string>

Go to the source code of this file.

Namespaces

namespace  popops
 Common functions, such as elementwise and reductions.
 

Functions

poplar::Tensor popops::sort (poplar::Graph &graph, const poplar::Tensor &t, unsigned dim, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
 
void popops::sortInPlace (poplar::Graph &graph, const poplar::Tensor &t, unsigned dim, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
 
poplar::Tensor popops::sortKeyValue (poplar::Graph &graph, const poplar::Tensor &k, const poplar::Tensor &v, unsigned dim, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
 
void popops::sortKeyValueInPlace (poplar::Graph &graph, const poplar::Tensor &k, const poplar::Tensor &v, unsigned dim, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={})
 

Detailed Description

Functions for sorting tensors.