Home
last modified time | relevance | path

Searched defs:gpu_options (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator.cc22 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()
82 const GPUOptions& gpu_options, in GPUBFCAllocator()
Dgpu_event_mgr.cc92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
304 const GPUOptions& gpu_options) { in GetEventMgr()
Dgpu_device.cc770 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()
845 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()
985 const auto& gpu_options = options.config.gpu_options(); in CreateDevices() local
Dgpu_device_test.cc74 GPUOptions* gpu_options = config->mutable_gpu_options(); local
Dgpu_event_mgr_test.cc42 TEST_EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in TEST_EventMgr()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc89 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
Dconvert_graph.cc566 GPUOptions gpu_options; in GetDeviceAndAllocator() local
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto437 GPUOptions gpu_options = 6; field
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc131 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig() local