Internal (Oneflow only)