Home
last modified time | relevance | path

Searched defs:gpu_ids (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_gpu_device.cc119 std::optional<std::set<int>> gpu_ids = in CreateDevices() local
H A Dxla_platform_info.cc27 std::set<int> gpu_ids; in ParseVisibleDeviceList() local