HomeSort by relevance Sort by last modified time
    Searched defs:gpu_manager (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/grappler/
devices.cc 33 perftools::gputools::Platform* gpu_manager = GPUMachineManager(); local
34 if (gpu_manager != nullptr) {
35 int num_gpus = gpu_manager->VisibleDeviceCount();
37 auto exec_status = gpu_manager->ExecutorForDevice(i);
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 668 gpu::Platform* gpu_manager = GPUMachineManager(); local
673 visible_gpu_order->resize(gpu_manager->VisibleDeviceCount());
688 if (cuda_gpu_id < 0 || cuda_gpu_id >= gpu_manager->VisibleDeviceCount()) {
692 gpu_manager->VisibleDeviceCount());
850 gpu::Platform* gpu_manager = GPUMachineManager(); local
1297 gpu::Platform* gpu_manager = GPUMachineManager(); local
    [all...]

Completed in 74 milliseconds