The DataGrid Search As You Type feature provides search text highlighting through the CellContentStyle property of the columns.
CellContentStyle is taken into account only for the default columns, so when you specify a custom CellContentTemplate or use a TemplateColumn currently the search text highlighting is not applied.