Poplar and PopLibs
|
Create tensor layouts that are optimised for host transfers. More...
#include <poplar/Tensor.hpp>
Go to the source code of this file.
Classes | |
struct | popops::IndicesAndTensor |
The pair of values returned by createHostSliceableTensor(). More... | |
Namespaces | |
namespace | poplar |
Poplar classes and functions. | |
namespace | popops |
Common functions, such as elementwise and reductions. | |
Functions | |
IndicesAndTensor | popops::createHostSliceableTensor (poplar::Graph &graph, const poplar::Type &type, const std::vector< size_t > &shape, const bool isRead, const poplar::DebugContext &debugContext={}) |
Create a Tensor that is well laid out for a host exchange copy and at the same time create the index tensor for the copy. More... | |
poplar::Tensor | popops::createHostTransferableTensor (poplar::Graph &graph, const poplar::Type &type, const std::vector< size_t > &shape, bool isRead, const poplar::DebugContext &debugContext={}) |
Create a tensor that is well laid out for a host exchange copy. More... | |
Create tensor layouts that are optimised for host transfers.