OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:visible_gpu_order
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc
666
std::vector<CudaGpuId>*
visible_gpu_order
) {
667
visible_gpu_order
->clear();
673
visible_gpu_order
->resize(gpu_manager->VisibleDeviceCount());
676
std::generate(
visible_gpu_order
->begin(),
visible_gpu_order
->end(),
694
visible_gpu_order
->push_back(CudaGpuId(cuda_gpu_id));
699
std::set<CudaGpuId> visible_device_set(
visible_gpu_order
->begin(),
700
visible_gpu_order
->end());
701
if (visible_device_set.size() !=
visible_gpu_order
->size()) {
711
const std::vector<CudaGpuId>&
visible_gpu_order
,
865
std::vector<CudaGpuId>
visible_gpu_order
;
local
[
all
...]
gpu_device.h
161
const std::vector<CudaGpuId>&
visible_gpu_order
,
195
// based upon '
visible_gpu_order
' which was generated by parsing
198
Status GetValidDeviceIds(const std::vector<CudaGpuId>&
visible_gpu_order
,
Completed in 317 milliseconds