Inheritance diagram for torch::autograd::python::PythonEngine:

Public Member Functions
void	thread_init (int device) override

void	thread_on_exception (FunctionTask &task, std::exception &e) override

variable_list	execute (const edge_list &roots, const variable_list &inputs, bool keep_graph, bool create_graph, const edge_list &outputs={}) override

std::unique_ptr< AnomalyMetadata >	make_anomaly_metadata () override

Public Member Functions inherited from torch::autograd::Engine
void	queue_callback (std::function< void()> callback)

bool	is_checkpoint_valid ()

Additional Inherited Members
Public Types inherited from torch::autograd::Engine
using	ready_queue_type = std::deque< std::pair< std::shared_ptr< Function >, InputBuffer >>

using	dependencies_type = std::unordered_map< Function *, int >

Static Public Member Functions inherited from torch::autograd::Engine
static Engine &	get_default_engine ()
	Returns a reference to a static `Engine` instance.

Protected Member Functions inherited from torch::autograd::Engine
void	compute_dependencies (Function *root, GraphTask &task)

void	evaluate_function (FunctionTask &task)

ReadyQueue &	ready_queue (at::Device device)

ReadyQueue &	ready_queue_by_index (int device_index)

void	start_threads ()

virtual void	thread_main (GraphTask *graph_task)

Protected Attributes inherited from torch::autograd::Engine
std::once_flag	start_threads_flag

std::vector< std::shared_ptr< ReadyQueue > >	ready_queues

std::vector< std::function< void()> >	final_callbacks

std::mutex	post_callbacks_lock

Detailed Description

Definition at line 13 of file python_engine.h.

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

Public Member Functions