Searched refs:merge_var (Results 1 – 1 of 1) sorted by relevance
521 def _EnforceShapeInvariant(merge_var, next_var): argument534 if isinstance(merge_var, ops.Tensor):535 m_shape = merge_var.get_shape()538 enter = merge_var.op.inputs[0].op547 raise TypeError("Type %s not supported" % type(merge_var))