Searched defs:gpu_options (Results 1 – 15 of 15) sorted by relevance
23 const GPUOptions& gpu_options) { in GetAllowGrowthValue()84 const GPUOptions& gpu_options, const string& name) in PluggableDeviceBFCAllocator()
22 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()84 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()
888 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()1016 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()1216 const auto& gpu_options = options.config.gpu_options(); in CreateDevices() local
73 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()
117 GPUOptions gpu_options; in Acquire() local
90 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
658 GPUOptions gpu_options; in GetDeviceAndAllocator() local
476 GPUOptions gpu_options = 6; field
327 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable() local
475 tensorflow::GPUOptions* gpu_options = in GetMultiClientServerDef() local
205 const GpuExecutableRunOptions* gpu_options = in ExecuteThunks() local
141 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig() local