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: