2da86542c46c23ea1ee7e2e214eb685349a6bb75,official/nlp/modeling/ops/decoding_module.py,,shape_list,#Any#,65

Before Change

def shape_list(tensor):
  Return a list of the tensor"s shape, and ensure no None values in list.
  // Get statically known shape (may contain None"s for unknown dimensions)
  shape = tensor.get_shape().as_list()

  // Ensure that the shape values are not None
  dynamic_shape = tf.shape(tensor)
  for i in range(len(shape)):  // pylint: disable=consider-using-enumerate
    if shape[i] is None:
      shape[i] = dynamic_shape[i]
  return shape


def get_shape_keep_last_dim(tensor):

After Change

def shape_list(tensor):
  Return a list of the tensor"s shape, and ensure no None values in list.
  return tf_utils.get_shape_list(tensor)


def get_shape_keep_last_dim(tensor):
  shape_list_obj = shape_list(tensor)

Instances