Poplar and PopLibs
|
Support for dynamic slices. More...
#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>
#include <poputil/DebugInfo.hpp>
#include <string>
#include <vector>
Go to the source code of this file.
Namespaces | |
namespace | poplar |
Poplar classes and functions. | |
namespace | popops |
Common functions, such as elementwise and reductions. | |
Functions | |
void | popops::sequenceSlice (poplar::Graph &graph, const poplar::Tensor &tIn, const poplar::Tensor &tOut, const poplar::Tensor &tN, const poplar::Tensor &tInOffset, const poplar::Tensor &tOutOffset, bool zeroUnused, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}) |
Slice a 2d tensor based on offsets specified by a tensor. More... | |
Support for dynamic slices.