HomeSort by relevance Sort by last modified time
    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