tensor_pool.h File Reference

#include "nndeploy/base/any.h"
#include "nndeploy/base/common.h"
#include "nndeploy/base/glic_stl_include.h"
#include "nndeploy/base/log.h"
#include "nndeploy/base/macro.h"
#include "nndeploy/base/object.h"
#include "nndeploy/base/status.h"
#include "nndeploy/base/string.h"
#include "nndeploy/net/util.h"

Include dependency graph for tensor_pool.h:

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

Go to the source code of this file.

Classes
struct	nndeploy::net::Chunk
struct	nndeploy::net::Offset
struct	nndeploy::net::OpBreadth
class	nndeploy::net::TensorPool
class	nndeploy::net::TensorPoolCreator
	TensorPool的创建类 More...
struct	nndeploy::net::TensorUsageRecord
class	nndeploy::net::TypeTensorPoolCreator< T >
	TensorPool的创建类模板 More...
class	nndeploy::net::TypeTensorPoolRegister< T >
	TensorPool的创建类的注册类模板 More...

Namespaces
	nndeploy
	nndeploy::net

Enumerations
enum	nndeploy::net::TensorPoolType : int {   nndeploy::net::kTensorPool1DSharedObjectTypeGreedyByBreadth , nndeploy::net::kTensorPool1DSharedObjectTypeGreedyBySize , nndeploy::net::kTensorPool1DSharedObjectTypeGreedyBySizeImprove , nndeploy::net::kTensorPool1DOffsetCalculateTypeGreedyBySize ,   nndeploy::net::kTensorPool1DOffsetCalculateTypeGreedyByBreadth , nndeploy::net::kTensorPool1DNone }

Functions
void	nndeploy::net::chunkPrint (const std::vector< std::shared_ptr< Chunk >> &chunks)
TensorPool *	nndeploy::net::createTensorPool (TensorPoolType type, device::Device *device, std::vector< TensorWrapper * > &tensor_repository, std::vector< OpWrapper * > &op_repository)
	Create a TensorPool object. More...
std::map< TensorPoolType, std::shared_ptr< TensorPoolCreator > > &	nndeploy::net::getGlobalTensorPoolCreatorMap ()
	Get the Global TensorPool Creator Map object. More...
std::vector< int >	nndeploy::net::getOpOrderIndex (std::vector< OpWrapper * > &producers, std::vector< OpWrapper * > &consumers, std::vector< OpWrapper * > &op_repository)
bool	nndeploy::net::isInterval (std::array< int, 2 > &interval, std::vector< std::array< int, 2 >> &intervals)
TensorPoolType	nndeploy::net::stringToTensorPoolType (const std::string &src)
std::string	nndeploy::net::tensorPoolTypeToString (TensorPoolType type)
void	nndeploy::net::tensorUsageRecordPrint (const std::vector< std::shared_ptr< TensorUsageRecord >> &tensor_usage_records)