Searched defs:GraphTask (Results 1 – 2 of 2) sorted by relevance
18 struct GraphTask : std::enable_shared_from_this<GraphTask> { struct37 // Exec info is created for each GraphTask, which allows filtering paths on argument55 int output_idx_; // within the output vector of a GraphTask argument116 // execution of the GraphTask is completed, the captured_vars_ are moved argument126 // Per-device current streams of the execute() that called this GraphTask. argument151 // check if the GraphTask is completed or not argument174 // GraphTask so that we know which ready queue we should push to if we are on argument182 // Final callbacks installed during execution of this GraphTask argument192 GraphTask( argument
658 GraphTask::GraphTask( in GraphTask() function in torch::autograd::GraphTask