SparseTensor.hpp

Go to the documentation of this file.

// Copyright (c) 2020 Graphcore Ltd. All rights reserved.
#ifndef popsparse_SparseTensor_hpp
#define popsparse_SparseTensor_hpp
 
#include <poplar/Tensor.hpp>
 
#include <poputil/TensorMetaData.hpp>
 
namespace popsparse {
namespace dynamic {
 
class SparseTensor {
  poplar::Tensor metaInfo;
  poplar::Tensor nzValues;
 
  poputil::TensorMetaData opMetaData;
 
public:
  SparseTensor() = default;
 
  SparseTensor(const SparseTensor &t) = default;
 
  SparseTensor(const poplar::Tensor &metaInfo, const poplar::Tensor &nzValues,
               const poputil::TensorMetaData &opMetaData = {})
      : metaInfo(metaInfo), nzValues(nzValues), opMetaData(opMetaData) {}
 
  const poplar::Tensor &getMetaInfoTensor() const { return metaInfo; }
 
  const poplar::Tensor &getNzValuesTensor() const { return nzValues; }
 
  const poputil::TensorMetaData &getOpMetaData() const { return opMetaData; }
};
 
} // end namespace dynamic
} // end namespace popsparse
 
#endif // popsparse_SparseTensor_hpp