#include <layer.hpp>

Public Member Functions
bool	operator== (const ParallelStrategy &ps) const

bool	operator!= (const ParallelStrategy &ps) const

Public Attributes
int	sample_groups = 0

int	sample_splits = 0

int	depth_groups = 0

int	depth_splits = 0

int	height_groups = 0

int	height_splits = 0

int	width_groups = 0

int	width_splits = 0

int	channel_groups = 0

int	channel_splits = 0

int	filter_groups = 0

int	filter_splits = 0

int	replications = 0

bool	enable_subgraph = false

int	sub_branch_tag = 0

int	sub_branch_resource_percentage = 0

Detailed Description

Represents a parallel strategy for a layer.

Definition at line 136 of file layer.hpp.

Member Function Documentation

◆ operator!=()

bool lbann::ParallelStrategy::operator!= ( const ParallelStrategy & ps ) const

inline

Definition at line 188 of file layer.hpp.

◆ operator==()

bool lbann::ParallelStrategy::operator== ( const ParallelStrategy & ps ) const

inline

Definition at line 170 of file layer.hpp.

Member Data Documentation

◆ channel_groups

int lbann::ParallelStrategy::channel_groups = 0

Number of process groups the channel dimension is split over.

Definition at line 155 of file layer.hpp.

◆ channel_splits

int lbann::ParallelStrategy::channel_splits = 0

Number of groups the channel dimension is split over.

Definition at line 157 of file layer.hpp.

◆ depth_groups

int lbann::ParallelStrategy::depth_groups = 0

Number of process groups the depth dimension is split over.

Definition at line 143 of file layer.hpp.

◆ depth_splits

int lbann::ParallelStrategy::depth_splits = 0

Number of groups the depth dimension is split over.

Definition at line 145 of file layer.hpp.

◆ enable_subgraph

bool lbann::ParallelStrategy::enable_subgraph = false

Enable subgraph for the layer.

Definition at line 165 of file layer.hpp.

◆ filter_groups

int lbann::ParallelStrategy::filter_groups = 0

Number of process groups the filter dimension is split over.

Definition at line 159 of file layer.hpp.

◆ filter_splits

int lbann::ParallelStrategy::filter_splits = 0

Number of groups the filter dimension is split over.

Definition at line 161 of file layer.hpp.

◆ height_groups

int lbann::ParallelStrategy::height_groups = 0

Number of process groups the height dimension is split over.

Definition at line 147 of file layer.hpp.

◆ height_splits

int lbann::ParallelStrategy::height_splits = 0

Number of groups the height dimension is split over.

Definition at line 149 of file layer.hpp.

◆ replications

int lbann::ParallelStrategy::replications = 0

Number of times the layer is replicated (for FC layers right now).

Definition at line 163 of file layer.hpp.

◆ sample_groups

int lbann::ParallelStrategy::sample_groups = 0

Number of process groups the sample dimension is split over.

Definition at line 139 of file layer.hpp.

◆ sample_splits

int lbann::ParallelStrategy::sample_splits = 0

Number of groups the sample dimension is split over.

Definition at line 141 of file layer.hpp.

◆ sub_branch_resource_percentage

int lbann::ParallelStrategy::sub_branch_resource_percentage = 0

percentage of parent resources to be allocated to this branch.

Definition at line 169 of file layer.hpp.

◆ sub_branch_tag

int lbann::ParallelStrategy::sub_branch_tag = 0

Branch number in the sub graph.

Definition at line 167 of file layer.hpp.

◆ width_groups

int lbann::ParallelStrategy::width_groups = 0

Number of process groups the width dimension is split over.

Definition at line 151 of file layer.hpp.

◆ width_splits

int lbann::ParallelStrategy::width_splits = 0

Number of groups the width dimension is split over.

Definition at line 153 of file layer.hpp.

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

layer.hpp

Public Member Functions

Public Attributes