repack_HWC_to_CHW_layout.hpp File Reference

#include "lbann/transforms/transform.hpp"

Include dependency graph for repack_HWC_to_CHW_layout.hpp:

Go to the source code of this file.

Classes
class	lbann::transform::repack_HWC_to_CHW_layout

Namespaces
	lbann
	lbann::transform

Functions
template<typename T >
void	lbann::transform::repack_HWC_to_CHW (T const *hwc_src, T *chw_dest, std::vector< size_t > const &chw_dims)
template<typename T >
void	lbann::transform::repack_DHWC_to_CDHW (T const *dhwc_src, T *cdhw_dest, std::vector< size_t > const &cdhw_dims)