Searched defs:graphTask (Results 1 – 2 of 2) sorted by relevance
89 auto graphTask = task.base_.lock(); in globalCpuThread() local200 auto graphTask = std::make_shared<GraphTask>( in computeDependencies() local412 auto graphTask = autogradContext->retrieveGraphTask(); in runEngineAndAccumulateGradients() local492 auto graphTask = autogradContext->retrieveGraphTask(); in executeSendFunctionAsync() local548 auto graphTask = autogradContext->retrieveGraphTask(); in executeSendFunctionAsync() local
122 std::shared_ptr<torch::autograd::GraphTask> graphTask) { in setGraphTask()