Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::jit::script::Method Struct Reference

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< Graphgraph_for (Stack inputs)
 
TORCH_API std::shared_ptr< Graphgraph () const
 
TORCH_API const std::string & name () const
 
Valueemit_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 Valueget_or_add_parameter (IValue *slot)
 
TORCH_API Valueget_or_add_attribute (TypePtr type, IValue *slot)
 
std::shared_ptr< Graphpropagate_shapes (std::vector< at::Tensor > inputs, bool with_grad=false)
 
std::shared_ptr< Graphpropagate_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
 
MethodsetSchema (FunctionSchema schema_)
 
TORCH_API const FunctionSchemagetSchema () const
 
std::string pretty_print_schema () const
 
GraphExecutorState getDebugState ()
 
void debugDisableAutodiffSubgraphInlining ()
 
bool is_optimized () const
 
Moduleowner () 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: