Loading...
Searching...
No Matches
Public Member Functions |
Static Public Member Functions |
Protected Member Functions |
List of all members
FromTorchScript< T > Class Template Reference
Read the tensors provided by a torch script file. More...
Detailed Description
template<class T>
class neml2::FromTorchScript< T >
class neml2::FromTorchScript< T >
Read the tensors provided by a torch script file.
#include <FromTorchScript.h>
Inheritance diagram for FromTorchScript< T >:
Public Member Functions | |
| FromTorchScript (const OptionSet &options) | |
 Public Member Functions inherited from UserTensorBase< T > | |
| UserTensorBase (const OptionSet &options) | |
| void | setup () override |
| Setup this object. | |
| Public Member Functions inherited from NEML2Object | |
| NEML2Object ()=delete | |
| NEML2Object (NEML2Object &&)=delete | |
| NEML2Object (const NEML2Object &)=delete | |
| NEML2Object & | operator= (NEML2Object &&)=delete |
| NEML2Object & | operator= (const NEML2Object &)=delete |
| virtual | ~NEML2Object ()=default |
| NEML2Object (const OptionSet &options) | |
| Construct a new NEML2Object object. | |
| const OptionSet & | input_options () const |
| const std::string & | name () const |
| A readonly reference to the object's name. | |
| const std::string & | type () const |
| A readonly reference to the object's type. | |
| const std::string & | path () const |
| A readonly reference to the object's path. | |
| const std::string & | doc () const |
| A readonly reference to the object's docstring. | |
| Factory * | factory () const |
| Get the factory that created this object. | |
| const Settings & | settings () const |
| Settings. | |
| template<typename T = NEML2Object> | |
| const T * | host () const |
| Get a readonly pointer to the host. | |
| template<typename T = NEML2Object> | |
| T * | host () |
| Get a writable pointer to the host. | |
| template<typename T > | |
| const T & | resolve_tensor (const std::string &name) |
| Resolve a TensorName to a Tensor. | |
| template<class T > | |
| std::shared_ptr< T > | get_object (const std::string §ion, const std::string &name) |
| template<class T = Solver> | |
| std::shared_ptr< T > | get_solver (const std::string &name) |
| Get a solver from the factory. | |
| template<class T = Data> | |
| std::shared_ptr< T > | get_data (const std::string &name) |
| Get a data from the factory. | |
| template<class T = Model> | |
| std::shared_ptr< T > | get_model (const std::string &name) |
| Get a model from the factory. | |
| template<class T = Driver> | |
| std::shared_ptr< T > | get_driver (const std::string &name) |
| Get a driver from the factory. | |
| template<class T = WorkScheduler> | |
| std::shared_ptr< T > | get_scheduler (const std::string &name) |
| Get a scheduler from the factory. | |
Static Public Member Functions | |
| static OptionSet | expected_options () |
| Static Public Member Functions inherited from UserTensorBase< T > | |
| static OptionSet | expected_options () |
| Static Public Member Functions inherited from NEML2Object | |
| static OptionSet | expected_options () |
Protected Member Functions | |
| T | make () const override |
| Helper to make the tensor given user input options. | |
| Protected Member Functions inherited from UserTensorBase< T > | |
Additional Inherited Members | |
| Static Protected Member Functions inherited from UserTensorBase< T > | |
| static std::string | tensor_type () |
| A human-readable description of the tensor type. | |
Constructor & Destructor Documentation
◆ FromTorchScript()
template<class T >
| FromTorchScript | ( | const OptionSet & | options | ) |
Member Function Documentation
◆ expected_options()
template<class T >
|
static |
◆ make()
template<class T >
|
overrideprotectedvirtual |
Helper to make the tensor given user input options.
Implements UserTensorBase< T >.
