tensor_types.h File Reference

This file contains the definition of the DataType enum class. More...

#include <regex>
#include <string>
#include <vector>
#include <numeric>
#include <complex>
#include <utility>
#include <iomanip>
#include <iostream>
#include <stdexcept>
#include <algorithm>
#include <unordered_map>
#include <initializer_list>
#include "source_base/module_device/types.h"

Include dependency graph for tensor_types.h:

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

Go to the source code of this file.

Classes
struct	container::DEVICE_CPU
	A tag type for identifying CPU and GPU devices. More...
struct	container::DEVICE_GPU
struct	container::GetTypeReal< T >
	Template struct to determine the return type based on the input type. More...
struct	container::GetTypeReal< std::complex< float > >
	Specialization of GetTypeReal for std::complex<float>. More...
struct	container::GetTypeReal< std::complex< double > >
	Specialization of GetTypeReal for std::complex<double>. More...
struct	container::PsiToContainer< T >
struct	container::PsiToContainer< base_device::DEVICE_CPU >
struct	container::PsiToContainer< base_device::DEVICE_GPU >
struct	container::ContainerToPsi< T >
struct	container::ContainerToPsi< container::DEVICE_CPU >
struct	container::ContainerToPsi< container::DEVICE_GPU >
struct	container::DeviceTypeToEnum< T >
	Template struct for mapping a Device Type to its corresponding enum value. More...
struct	container::DeviceTypeToEnum< DEVICE_CPU >
struct	container::DeviceTypeToEnum< DEVICE_GPU >
struct	container::DeviceTypeToEnum< base_device::DEVICE_CPU >
struct	container::DeviceTypeToEnum< base_device::DEVICE_GPU >
struct	container::DataTypeToEnum< T >
	Template struct for mapping a DataType to its corresponding enum value. More...
struct	container::DataTypeToEnum< int >
struct	container::DataTypeToEnum< float >
struct	container::DataTypeToEnum< double >
struct	container::DataTypeToEnum< int64_t >
struct	container::DataTypeToEnum< std::complex< float > >
struct	container::DataTypeToEnum< std::complex< double > >

Namespaces
namespace	container

Enumerations
enum class	container::DataType {   container::DT_INVALID = 0 , container::DT_FLOAT = 1 , container::DT_DOUBLE = 2 , container::DT_INT = 3 ,   container::DT_INT64 = 4 , container::DT_COMPLEX = 5 , container::DT_COMPLEX_DOUBLE = 6 }
	Enumeration of data types for tensors. The DataType enum lists the supported data types for tensors. Each data type is identified by a unique value. The DT_INVALID value is reserved for invalid data types. More...
enum class	container::DeviceType { container::UnKnown = 0 , container::CpuDevice = 1 , container::GpuDevice = 2 }
	The type of memory used by an allocator. More...

Functions
std::ostream &	container::operator<< (std::ostream &os, const DataType &data_type)
	Overloaded operator<< for the Tensor class.
std::ostream &	container::operator<< (std::ostream &os, const DeviceType &memory_type)
	Overloaded operator<< for the Tensor class.

Detailed Description

This file contains the definition of the DataType enum class.