cnpy_utils.hpp File Reference

#include "cnpy.h"
#include "lbann/utils/exception.hpp"
#include <string>
#include <vector>

Include dependency graph for cnpy_utils.hpp:

Go to the source code of this file.

Namespaces
	lbann
	lbann::cnpy_utils

Functions
size_t	lbann::cnpy_utils::compute_cnpy_array_offset (const cnpy::NpyArray &na, std::vector< size_t > indices)
template<typename T >
size_t	lbann::cnpy_utils::ptr_offset (const cnpy::NpyArray &na, std::vector< size_t > indices)
template<typename T >
T &	lbann::cnpy_utils::data (const cnpy::NpyArray &na, const std::vector< size_t > indices)
template<typename T >
T *	lbann::cnpy_utils::data_ptr (const cnpy::NpyArray &na, const std::vector< size_t > indices)
template<>
void *	lbann::cnpy_utils::data_ptr< void > (const cnpy::NpyArray &na, const std::vector< size_t > indices)
void	lbann::cnpy_utils::shrink_to_fit (cnpy::NpyArray &na, size_t sz)
std::string	lbann::cnpy_utils::show_shape (const cnpy::NpyArray &na)
	Show the dimensions of loaded data. More...