Searched refs:shape_like (Results 1 – 1 of 1) sorted by relevance
38 def partial_shape_to_tensor(shape_like): argument53 shape_like = tensor_shape.as_shape(shape_like)55 [dim if dim is not None else -1 for dim in shape_like.as_list()],61 ret = ops.convert_to_tensor(shape_like, preferred_dtype=dtypes.int64)65 % (shape_like, ret.shape))69 % (shape_like, ret.dtype.name))