latest/doxygen/SequenceSlice_8hpp_source.html

// Copyright (c) 2021 Graphcore Ltd. All rights reserved.

#ifndef popops_SequenceSlice_hpp

#define popops_SequenceSlice_hpp

#include <poplar/Graph.hpp>

#include <poplar/Program.hpp>

#include <poputil/DebugInfo.hpp>

#include <string>

#include <vector>


namespace poplar {

class Tensor;

}


namespace popops {


void 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 = {});


} // end namespace popops

#endif // popops_DynamicSlice_hpp

DebugInfo.hpp
Poplibs generic debug info structure.

poplar::DebugContext
DebugContext gathers the common external parameters of the context of an operation.
Definition: DebugContext.hpp:221

poplar::Graph
This class represents a graph program to be executed on the IPU.
Definition: Graph.hpp:52

poplar::Tensor
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38

poplar::program::Sequence
Program that executes a sequence of programs.
Definition: Program.hpp:77

poplar
Poplar classes and functions.
Definition: ArrayRef.hpp:14

popops
Common functions, such as elementwise and reductions.
Definition: AllTrue.hpp:15

popops::sequenceSlice
void 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.