Searched refs:merge_var (Results 1 – 1 of 1) sorted by relevance
519 def _EnforceShapeInvariant(merge_var, next_var): argument532 if isinstance(merge_var, ops.Tensor):533 m_shape = merge_var.get_shape()536 enter = merge_var.op.inputs[0].op