weights_helpers.hpp File Reference

#include "lbann/utils/exception.hpp"
#include "lbann/utils/typename.hpp"
#include "lbann/weights/data_type_weights.hpp"
#include "lbann/weights/weights.hpp"

Include dependency graph for weights_helpers.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	lbann::weights_details::SafeWeightsAccessor< TensorDataType >
	Ensure safe access to weights objects' data. More...

Namespaces
	lbann
	lbann::weights_details

Detailed Description

A hacky utility for dealing with layers that require access to mutable weights values (e.g. batchnorm layers). A future refactor should focus on cleaning this up; a suitable metric for at least some success would be the deletion of this file.

This file mostly exists because the code is common across several layers, but it is not needed in the header.

Definition in file weights_helpers.hpp.