OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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