DataGrid - How to set focus to a column?

I wonder if it is possible to set the focus to a data grid after adding a new row. In my case setting focus to product code column.

How to set focus to ProductCode in OnAfterRenderAsync?

protected override Task OnAfterRenderAsync(bool firstRender)

<RadzenDataGridColumn TItem="OrderDetailDto" Property="ProductCode" Title="Product  Code" OrderIndex="2">
    <EditTemplate Context="orderDetail">
        <RadzenTextBox @bind-Value="orderDetail.ProductCode" Style="width: 100%; display: block" Name="ProductCode" />
        <RadzenRequiredValidator Text="Product Code is Required!" Component="ProductCode" Popup="true" Style="position: absolute; z-index: 9999;"/>