tfma.types.is_tensor