Replace using_pjrt()
xla runtime device_type()
check with in xla.…
#4578
Loading
using_pjrt()
xla runtime device_type()
check with in xla.…
#4578