AOTIModelHandle Member List

This is the complete list of members for AOTIModelHandle, including all inherited members.

_m	AOTIModelHandle	private
AOTIModelHandle(const std::string &input, const std::string &model_name, const std::vector< std::string > &devices, const std::vector< std::size_t > &batch_sizes, const void *comm)	AOTIModelHandle	inline
device() const override	AOTIModelHandle	inlinevirtual
input_base_shapes() const override	AOTIModelHandle	inlinevirtual
input_names() const override	AOTIModelHandle	inlinevirtual
jacobian(const std::map< std::string, at::Tensor > &inputs) const override	AOTIModelHandle	inlinevirtual
load(const std::string &input, const std::string &model_name, const std::vector< std::string > &devices, const std::vector< std::size_t > &batch_sizes, const void *comm)	AOTIModelHandle	inlineprivatestatic
output_base_shapes() const override	AOTIModelHandle	inlinevirtual
output_names() const override	AOTIModelHandle	inlinevirtual
param_jacobian(const std::map< std::string, at::Tensor > &inputs) const override	AOTIModelHandle	inlinevirtual
parameter_base_shapes() const override	AOTIModelHandle	inlinevirtual
set_parameter(const std::string &name, const at::Tensor &value) override	AOTIModelHandle	inlinevirtual
value(const std::map< std::string, at::Tensor > &inputs) const override	AOTIModelHandle	inlinevirtual
~NEML2ModelHandle()=default	NEML2ModelHandle	virtual