NEML2Utils.C File Reference

Go to the source code of this file.

Namespaces
namespace	NEML2Utils

Functions
void	NEML2Utils::dumpInputsToTorchScript (const std::map< std::string, at::Tensor > &inputs, const std::string &filename)
	Serialize a set of named NEML2 input tensors to a TorchScript archive at filename (a torch::jit::Module whose buffers are the input tensors), loadable offline in Python via torch.jit.load.
void	NEML2Utils::redirectLogsToConsole (const ConsoleStream &console)
	Route NEML2's log output into MOOSE's console.
std::string	NEML2Utils::stringify (MOOSEIOType type)
std::string	NEML2Utils::docstring (const std::string &desc)
	Augment docstring if NEML2 is not enabled.
void	NEML2Utils::assertNEML2Enabled ()
	Assert that NEML2 is enabled.

Variables
static const std::string	NEML2Utils::missing_neml2