OpenNN
Open-source neural networks library
Loading...
Searching...
No Matches
tensor_utilities.h File Reference
#include "pch.h"
#include "configuration.h"

Go to the source code of this file.

Classes

struct  opennn::Shape
 
struct  opennn::Buffer
 
struct  opennn::TensorView
 
class  opennn::Backend
 

Namespaces

namespace  opennn
 

Typedefs

template<typename T, size_t N>
using opennn::array = Eigen::array<T, N>
 

Functions

int opennn::to_int (Index value)
 
float opennn::to_type (Index value)
 
Index opennn::align_up (Index value, Index alignment)
 
Index opennn::get_aligned_size (Index size)
 
Index opennn::get_aligned_bytes (Index n_bytes)
 
template<typename Container>
Index opennn::ssize (const Container &container) noexcept
 
bool opennn::is_aligned (const void *ptr)
 
Index opennn::aligned_total_elements (const vector< Shape > &shapes)
 
Index opennn::aligned_total_elements (const vector< vector< Shape > > &nested)
 
Index opennn::aligned_total_bytes (const vector< Shape > &shapes, const vector< Type > &dtypes)
 
Index opennn::aligned_total_bytes (const vector< vector< Shape > > &nested, const vector< vector< Type > > &dtypes)
 
Index opennn::aligned_total_bytes (const vector< Shape > &shapes, Type dtype)
 
string opennn::shape_to_string (const Shape &, const string &=" ")
 
Shape opennn::string_to_shape (const string &, const string &=" ")
 
template<typename... Vs>
size_t opennn::hash_combine (const Vs &... values)
 
ThreadPoolDevice & opennn::get_device ()
 

Variables

static constexpr Index opennn::ALIGN_BYTES = EIGEN_MAX_ALIGN_BYTES
 
static constexpr Index opennn::ALIGN_ELEMENTS = ALIGN_BYTES / sizeof(float)
 
constexpr cudaDataType_t opennn::CUDA_REDUCTION_DTYPE = CUDA_R_32F
 
constexpr cublasComputeType_t opennn::CUBLAS_COMPUTE_DTYPE = CUBLAS_COMPUTE_32F_FAST_TF32