lbann::tensor_overlap_constraints Class Reference

#include <distconv_adapter.hpp>

Public Types
using	dist_set = std::unordered_set< dc::Dist * >
using	const_dist_set = std::unordered_set< const dc::Dist * >

Public Member Functions
	tensor_overlap_constraints ()=default
virtual	~tensor_overlap_constraints ()=default
void	mark_equivalent (dc::Dist &d1, dc::Dist &d2)
void	mark_updated (const dc::Dist &d)
void	mark_invariant (const dc::Dist &d)
void	find_valid_overlap ()

Private Attributes
std::unordered_map< const dc::Dist *, dist_set >	m_equivalents
const_dist_set	m_updated
const_dist_set	m_invariants

Detailed Description

Definition at line 48 of file distconv_adapter.hpp.

Member Typedef Documentation

const_dist_set

using lbann::tensor_overlap_constraints::const_dist_set = std::unordered_set<const dc::Dist*>

Definition at line 52 of file distconv_adapter.hpp.

dist_set

using lbann::tensor_overlap_constraints::dist_set = std::unordered_set<dc::Dist*>

Definition at line 51 of file distconv_adapter.hpp.

Constructor & Destructor Documentation

tensor_overlap_constraints()

lbann::tensor_overlap_constraints::tensor_overlap_constraints ( )

default

~tensor_overlap_constraints()

virtual lbann::tensor_overlap_constraints::~tensor_overlap_constraints ( )

virtualdefault

Member Function Documentation

find_valid_overlap()

void lbann::tensor_overlap_constraints::find_valid_overlap

(

)

mark_equivalent()

void lbann::tensor_overlap_constraints::mark_equivalent	(	dc::Dist &	d1,
		dc::Dist &	d2
	)

Here is the caller graph for this function:

mark_invariant()

void lbann::tensor_overlap_constraints::mark_invariant

(

const dc::Dist &

d

)

Here is the caller graph for this function:

mark_updated()

void lbann::tensor_overlap_constraints::mark_updated

(

const dc::Dist &

d

)

Here is the caller graph for this function:

Member Data Documentation

m_equivalents

std::unordered_map<const dc::Dist*, dist_set> lbann::tensor_overlap_constraints::m_equivalents

private

Definition at line 64 of file distconv_adapter.hpp.

m_invariants

const_dist_set lbann::tensor_overlap_constraints::m_invariants

private

Definition at line 66 of file distconv_adapter.hpp.

m_updated

const_dist_set lbann::tensor_overlap_constraints::m_updated

private

Definition at line 65 of file distconv_adapter.hpp.

---

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

• distconv_adapter.hpp