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.