Searched refs:merge_var (Results 1 – 1 of 1) sorted by relevance
606 def _EnforceShapeInvariant(merge_var, next_var): argument619 if isinstance(merge_var, ops.Tensor):620 m_shape = merge_var.get_shape()630 (merge_var.name, m_shape, n_shape))635 m_values_shape = merge_var.values.get_shape()636 m_indices_shape = merge_var.indices.get_shape()638 if merge_var.dense_shape is not None:639 m_shape_shape = merge_var.dense_shape.get_shape()654 (merge_var.name, m_values_shape, m_indices_shape, m_shape_shape,657 m_values_shape = merge_var.values.get_shape()[all …]