Use unified function to check for TF backend

Related-Prod: PRODX-37464
Change-Id: Icad71d674f1c3460729a6945393858986d5799da
(cherry picked from commit c79954b8255ef40322c16c4bb6fad2ff4d31905d)
2 files changed