MLIR  20.0.0git
Public Member Functions | List of all members
mlir::sparse_tensor::MapRef Class Referencefinal

A class for capturing the sparse tensor type map with a compact encoding. More...

#include "mlir/ExecutionEngine/SparseTensor/MapRef.h"

Public Member Functions

 MapRef (uint64_t d, uint64_t l, const uint64_t *d2l, const uint64_t *l2d)
 
template<typename T >
void pushforward (const T *in, T *out) const
 
template<typename T >
void pushbackward (const T *in, T *out) const
 
uint64_t getDimRank () const
 
uint64_t getLvlRank () const
 

Detailed Description

A class for capturing the sparse tensor type map with a compact encoding.

Currently, the following situations are supported: (1) map is a permutation (2) map has affine ops (restricted set)

The pushforward/backward operations are fast for (1) but incur some obvious overhead for situation (2).

Definition at line 33 of file MapRef.h.

Constructor & Destructor Documentation

◆ MapRef()

mlir::sparse_tensor::MapRef::MapRef ( uint64_t  d,
uint64_t  l,
const uint64_t *  d2l,
const uint64_t *  l2d 
)

Definition at line 12 of file MapRef.cpp.

Member Function Documentation

◆ getDimRank()

uint64_t mlir::sparse_tensor::MapRef::getDimRank ( ) const
inline

Definition at line 82 of file MapRef.h.

◆ getLvlRank()

uint64_t mlir::sparse_tensor::MapRef::getLvlRank ( ) const
inline

Definition at line 83 of file MapRef.h.

◆ pushbackward()

template<typename T >
void mlir::sparse_tensor::MapRef::pushbackward ( const T *  in,
T *  out 
) const
inline

Definition at line 67 of file MapRef.h.

◆ pushforward()

template<typename T >
void mlir::sparse_tensor::MapRef::pushforward ( const T *  in,
T *  out 
) const
inline

Definition at line 43 of file MapRef.h.


The documentation for this class was generated from the following files: