Searched refs:mark_as_return (Results 1 – 3 of 3) sorted by relevance
50 val = c.mark_as_return(val)71 val = c.mark_as_return(val)93 one = c.mark_as_return(one)128 val = c.mark_as_return(val)151 val = c.mark_as_return(val)173 val = c.mark_as_return(val)195 val = c.mark_as_return(val)
115 def mark_as_return(self, tensor): member in AutomaticControlDependencies362 result_flat = [a.mark_as_return(t) for t in nest.flatten(result)]
662 x = a.mark_as_return(x)