model Class Reference

Ratpac-two: cppflow::model Class Reference
Ratpac-two
cppflow::model Class Reference

Public Types

enum  TYPE { SAVED_MODEL , FROZEN_GRAPH }
 

Public Member Functions

 model (const std::string &filename, const TYPE type=TYPE::SAVED_MODEL)
 
 model (const model &model)=default
 
 model (model &&model)=default
 
modeloperator= (const model &other)=default
 
modeloperator= (model &&other)=default
 
std::vector< tensoroperator() (std::vector< std::tuple< std::string, tensor >> inputs, std::vector< std::string > outputs)
 
tensor operator() (const tensor &input)
 
std::vector< std::string > get_operations () const
 
std::vector< int64_t > get_operation_shape (const std::string &operation) const
 

The documentation for this class was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/ratpac/checkouts/latest/src/external/cppflow/include/cppflow/model.h