Namespaces | Functions
tf_utils.H File Reference
Include dependency graph for tf_utils.H:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 tf_utils
 

Functions

TF_Graph * LoadGraph (const char *graphPath)
 
void DeleteGraph (TF_Graph *graph)
 
TF_Session * CreateSession (TF_Graph *graph)
 
void DeleteSession (TF_Session *session)
 
TF_Code RunSession (TF_Session *session, const TF_Output *inputs, TF_Tensor *const *input_tensors, std::size_t ninputs, const TF_Output *outputs, TF_Tensor **output_tensors, std::size_t noutputs)
 
TF_Code RunSession (TF_Session *session, const std::vector< TF_Output > &inputs, const std::vector< TF_Tensor * > &input_tensors, const std::vector< TF_Output > &outputs, std::vector< TF_Tensor * > &output_tensors)
 
TF_Tensor * CreateTensor (TF_DataType data_type, const std::int64_t *dims, std::size_t num_dims, const void *data, std::size_t len)
 
template<typename T >
TF_Tensor * CreateTensor (TF_DataType data_type, const std::vector< std::int64_t > &dims, const std::vector< T > &data)
 
TF_Tensor * CreateEmptyTensor (TF_DataType data_type, const std::int64_t *dims, std::size_t num_dims)
 
TF_Tensor * CreateEmptyTensor (TF_DataType data_type, const std::vector< std::int64_t > &dims)
 
void DeleteTensor (TF_Tensor *tensor)
 
void DeleteTensors (const std::vector< TF_Tensor * > &tensors)
 
void SetTensorsData (TF_Tensor *tensor, const void *data, std::size_t len)
 
template<typename T >
void SetTensorsData (TF_Tensor *tensor, const std::vector< T > &data)
 
template<typename T >
std::vector< TGetTensorsData (const TF_Tensor *tensor)
 
template<typename T >
std::vector< std::vector< T > > GetTensorsData (const std::vector< TF_Tensor * > &tensors)