nndeploy C++ API  0.2.0
nndeploy C++ API
Classes | Namespaces | Enumerations | Functions
util.h File Reference
#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/device/tensor.h"
#include "nndeploy/op/op.h"
Include dependency graph for util.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  nndeploy::net::OpWrapper
 
class  nndeploy::net::TensorWrapper
 

Namespaces

 nndeploy
 
 nndeploy::net
 

Enumerations

enum  nndeploy::net::InputOutputType { nndeploy::net::kInput = 0 , nndeploy::net::kOutput = 1 , nndeploy::net::kBoth = 2 , nndeploy::net::kNone = 3 }
 

Functions

bool nndeploy::net::checkTensor (const std::vector< device::Tensor * > &src_tensors, const std::vector< device::Tensor * > &dst_tensors)
 
std::vector< OpWrapper * > nndeploy::net::checkUnuseOp (std::vector< OpWrapper * > &op_repository)
 
std::vector< TensorWrapper * > nndeploy::net::checkUnuseTensor (std::vector< OpWrapper * > &op_repository, std::vector< TensorWrapper * > &tensor_repository)
 
base::Status nndeploy::net::dumpNet (std::vector< TensorWrapper * > &tensor_repository, std::vector< OpWrapper * > &op_repository, std::vector< device::Tensor * > &graph_inputs, std::vector< device::Tensor * > &graph_outputs, const std::string &name, std::ostream &oss)
 
std::vector< OpWrapper * > nndeploy::net::findEndOps (std::vector< OpWrapper * > &op_repository)
 
std::vector< TensorWrapper * > nndeploy::net::findEndTensors (std::vector< TensorWrapper * > &tensor_repository)
 
OpWrapper * nndeploy::net::findOpWrapper (std::vector< OpWrapper * > &op_repository, const std::string &name)
 
OpWrapper * nndeploy::net::findOpWrapper (std::vector< OpWrapper * > &op_repository, op::Op *node)
 
std::vector< OpWrapper * > nndeploy::net::findStartOps (std::vector< OpWrapper * > &op_repository)
 
std::vector< TensorWrapper * > nndeploy::net::findStartTensors (std::vector< TensorWrapper * > &tensor_repository)
 
TensorWrapper * nndeploy::net::findTensorWrapper (std::vector< TensorWrapper * > &tensor_repository, const std::string &tensor_name)
 
TensorWrapper * nndeploy::net::findTensorWrapper (std::vector< TensorWrapper * > &tensor_repository, device::Tensor *tensor)
 
op::Op * nndeploy::net::getOp (std::vector< OpWrapper * > &op_repository, const std::string &name)
 
device::Tensor * nndeploy::net::getTensor (std::vector< TensorWrapper * > &tensor_repository, const std::string &tensor_name)
 
template<typename T >
void nndeploy::net::insertUnique (std::vector< T > &vec, const T &val)
 对vector插入不在vector中的元素,即类似集合的作用 More...
 
void nndeploy::net::printNetInfo (const std::vector< OpWrapper * > &op_repository, const std::vector< TensorWrapper * > &tensor_repository)
 
base::Status nndeploy::net::setColor (std::vector< OpWrapper * > &op_repository, base::NodeColorType color)
 
base::Status nndeploy::net::topoSort (std::vector< OpWrapper * > &op_repository, base::TopoSortType topo_sort_type, std::vector< OpWrapper * > &topo_sort_node)
 
base::Status nndeploy::net::topoSortBFS (std::vector< OpWrapper * > &op_repository, std::vector< OpWrapper * > &topo_sort_node)
 
base::Status nndeploy::net::topoSortDFS (std::vector< OpWrapper * > &op_repository, std::vector< OpWrapper * > &topo_sort_node)