Searched defs:gpu_options (Results 1 – 12 of 12) sorted by relevance
22 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()82 const GPUOptions& gpu_options, in GPUBFCAllocator()
92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()251 const GPUOptions& gpu_options) { in GetEventMgr()
827 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()955 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()1161 const auto& gpu_options = options.config.gpu_options(); in CreateDevices() local
78 GPUOptions* gpu_options = config->mutable_gpu_options(); local
93 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()252 const GPUOptions& gpu_options) { in GetEventMgr()
42 TEST_EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in TEST_EventMgr()
90 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
643 GPUOptions gpu_options; in GetDeviceAndAllocator() local
184 const GpuExecutableRunOptions* gpu_options = in ExecuteThunks() local
449 GPUOptions gpu_options = 6; field
325 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable() local
140 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig() local