Public Member Functions
	Method (Module owner, std::string name, bool optimize, std::shared_ptr< Graph > graph, std::vector< IValue > initial_members, std::function< void(Method &)> method_creator)

void	run (Stack &stack)

void	run (Stack &&stack)

IValue	operator() (std::vector< IValue > stack)

std::shared_ptr< Graph >	graph_for (Stack inputs)

TORCH_API std::shared_ptr< Graph >	graph () const

TORCH_API const std::string &	name () const

Value *	emit_call_to (const SourceRange &loc, Method &callee, ArrayRef< NamedValue > args, ArrayRef< NamedValue > kwargs)

TORCH_API void	ensure_defined ()

size_t	num_inputs () const

TORCH_API Value *	get_or_add_parameter (IValue *slot)

TORCH_API Value *	get_or_add_attribute (TypePtr type, IValue *slot)

std::shared_ptr< Graph >	propagate_shapes (std::vector< at::Tensor > inputs, bool with_grad=false)

std::shared_ptr< Graph >	propagate_and_assign_input_and_output_shapes (std::vector< at::Tensor > inputs, std::vector< at::Tensor > outputs, bool with_grad=false, bool propagate=true)

const std::vector< IValue * > &	initial_ivalues () const

Method &	setSchema (FunctionSchema schema_)

TORCH_API const FunctionSchema &	getSchema () const

std::string	pretty_print_schema () const

GraphExecutorState	getDebugState ()

void	debugDisableAutodiffSubgraphInlining ()

bool	is_optimized () const

Module &	owner () const

void	check_single_output ()

Detailed Description

Definition at line 55 of file module.h.

The documentation for this struct was generated from the following files:

Public Member Functions