Searched refs:xla_test_device (Results 1 – 2 of 2) sorted by relevance
57 "--xla_test_device=XLA_CPU",71 "--xla_test_device=XLA_GPU",
54 string xla_test_device = *xla_test_device_ptr; in DeviceFromFlag() local55 return absl::StrCat("/device:", xla_test_device, ":0"); in DeviceFromFlag()