Searched refs:mark_as_return (Results 1 – 5 of 5) sorted by relevance
52 val = c.mark_as_return(val)113 c.mark_as_return(read_op.outputs[0])139 c.mark_as_return(read_op1.outputs[0])140 c.mark_as_return(read_op2.outputs[0])141 c.mark_as_return(read_op3.outputs[0])142 c.mark_as_return(read_op4.outputs[0])584 val = c.mark_as_return(val)606 one = c.mark_as_return(one)641 val = c.mark_as_return(val)664 val = c.mark_as_return(val)[all …]
192 def mark_as_return(self, tensor): member in AutomaticControlDependencies574 result_flat = [a.mark_as_return(t) for t in nest.flatten(result)]
962 x = deps_ctx.mark_as_return(x)
704 def mark_as_return(outputs, acd): function713 return_tensor = acd.mark_as_return(tensor)715 return_tensor._keras_mask = acd.mark_as_return(tensor._keras_mask)
107 return self.autodeps_scope.mark_as_return(t)