|
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< T > | GetTensorsData (const TF_Tensor *tensor) |
|
template<typename T > |
std::vector< std::vector< T > > | GetTensorsData (const std::vector< TF_Tensor * > &tensors) |
|