Searched refs:graphTask (Results 1 – 3 of 3) sorted by relevance
89 auto graphTask = task.base_.lock(); in globalCpuThread() local90 if (graphTask == nullptr) { in globalCpuThread()97 graphTask, in globalCpuThread()106 /*node_task*/ NodeTask(graphTask, graphRoot, std::move(inputs)), in globalCpuThread()200 auto graphTask = std::make_shared<GraphTask>( in computeDependencies() local220 graphTask->outstanding_tasks_++; in computeDependencies()228 auto& dependencies = graphTask->dependencies_; in computeDependencies()277 graphTask->stash_current_streams(); in computeDependencies()308 graphTask->init_to_execute( in computeDependencies()310 for (auto& mapEntry : graphTask->exec_info_) { in computeDependencies()[all …]
122 std::shared_ptr<torch::autograd::GraphTask> graphTask) { in setGraphTask() argument127 graphTask_ = std::move(graphTask); in setGraphTask()
98 void setGraphTask(std::shared_ptr<torch::autograd::GraphTask> graphTask);