|
Poplar and PopLibs
|
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={}) |
Functions for sorting tensors.