Searched defs:autogradContainer (Results 1 – 3 of 3) sorted by relevance
8 static DistAutogradContainer* autogradContainer = in getDistAutogradContainer() local
162 DistAutogradContainer* autogradContainer; variable
57 auto& autogradContainer = DistAutogradContainer::getInstance(); in addRecvRpcBackward() local107 auto& autogradContainer = DistAutogradContainer::getInstance(); in getMessageWithAutograd() local