Searched defs:mark_as_return (Results 1 – 2 of 2) sorted by relevance
192 def mark_as_return(self, tensor): member in AutomaticControlDependencies
704 def mark_as_return(outputs, acd): function