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

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

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.

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

inline

Definition at line 82 of file MapRef.h.

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

inline

Definition at line 83 of file MapRef.h.

template<typename T >

void mlir::sparse_tensor::MapRef::pushbackward	(	const T *	in,
		T *	out
	)		const

inline

Definition at line 67 of file MapRef.h.

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: