Caffe2 - C++ API
A deep learning, cross platform ML framework
Data Structures | Public Member Functions | Data Fields
torch::autograd::GraphTask Struct Reference

Data Structures

struct  ExecInfo
 

Public Member Functions

void init_to_execute (Function &graph_root, const edge_list &outputs)
 
bool can_checkpoint ()
 
 GraphTask (bool keep_graph, bool grad_mode)
 

Data Fields

std::exception_ptr exception
 
std::atomic_bool has_error
 
std::atomic< uint64_t > outstanding_tasks
 
bool keep_graph
 
bool grad_mode
 
std::mutex mutex
 
std::condition_variable not_done
 
std::unordered_map< Function *, InputBuffernot_ready
 
std::unordered_map< Function *, int > dependencies
 
std::unordered_map< Function *, ExecInfoexec_info
 
std::vector< Variablecaptured_vars
 
int owner
 

Detailed Description

Definition at line 128 of file engine.cpp.


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