Searched defs:retainGraph (Results 1 – 4 of 4) sorted by relevance
15 bool retainGraph) in RRefBackwardReq()55 bool retainGraph = tupleElements[2].toBool(); in fromMessage() local71 bool RRefBackwardReq::retainGraph() const { in retainGraph() function in torch::distributed::autograd::RRefBackwardReq
18 bool retainGraph) in PropagateGradientsReq()65 bool retainGraph = tupleElements.back().toBool(); in fromMessage() local94 bool PropagateGradientsReq::retainGraph() { in retainGraph() function in torch::distributed::autograd::PropagateGradientsReq
307 void PyRRef::backward(int64_t autogradContextId, bool retainGraph) { in backward()313 bool retainGraph, in backwardOwnerRRef()339 bool retainGraph, in backward()
186 bool retainGraph) { in computeDependencies()459 bool retainGraph) { in executeSendFunctionAsync()563 bool retainGraph) { in execute()