Home
last modified time | relevance | path

Searched defs:num_gpu (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/
H A Dxpu.cpp28 auto num_gpu = device_count(); in manual_seed_all() local
H A Dcuda.cpp43 auto num_gpu = device_count(); in manual_seed_all() local
/aosp_15_r20/device/google/cuttlefish/host/libs/vm_manager/
Dqemu_manager.cpp199 const int num_gpu = instance.hwcomposer() != kHwComposerNone; in ConfigureBootDevices() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc221 int num_gpu = ctx->local_device_mgr()->NumDeviceType(gpu_device_type); in InjectTfGpuResourcesHelper() local