Searched defs:gpu_options (Results 1 – 8 of 8) sorted by relevance
25 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()64 const GPUOptions& gpu_options, in GPUBFCAllocator()
92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
75 GPUOptions* gpu_options = config->mutable_gpu_options(); local
815 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()887 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()996 const auto& gpu_options = options.config.gpu_options(); in CreateDevices() local
163 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
745 GPUOptions gpu_options; in GetDeviceAndAllocator() local
380 GPUOptions gpu_options = 6; field
89 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig() local