| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-g-p-u-options.pbtxt | 1 path: "tensorflow.GPUOptions"
4 name: "GPUOptions"
58 type_name: ".tensorflow.GPUOptions.Experimental"
67 type_name: ".tensorflow.GPUOptions.Experimental.VirtualDevices"
|
| D | tensorflow.-config-proto.pbtxt | 54 type_name: ".tensorflow.GPUOptions"
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator.cc | 22 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()
78 : GPUBFCAllocator(sub_allocator, total_memory, GPUOptions(), name) {} in GPUBFCAllocator()
82 const GPUOptions& gpu_options, in GPUBFCAllocator()
|
| D | gpu_bfc_allocator.h | 40 const GPUOptions& gpu_options, const string& name);
46 static bool GetAllowGrowthValue(const GPUOptions& gpu_options);
|
| D | gpu_bfc_allocator_test.cc | 259 GPUOptions options; in TEST_P()
548 GPUOptions options; in TestForceAllowGrowth()
605 GPUOptions options; in TEST_F()
631 GPUOptions options; in TEST_F()
662 GPUOptions options; in TestRegionDeallocation()
709 GPUOptions options; in TestNoRegionDeallocation()
|
| D | gpu_event_mgr.cc | 92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
251 const GPUOptions& gpu_options) { in GetEventMgr()
|
| D | gpu_process_state.h | 84 virtual Allocator* GetGPUAllocator(const GPUOptions& options,
|
| D | gpu_process_state.cc | 84 const GPUOptions& options, PlatformGpuId platform_gpu_id, in CreateSubAllocator()
137 const GPUOptions& options, TfGpuId tf_gpu_id, size_t total_bytes, in GetGPUAllocator()
|
| D | gpu_device.cc | 242 const GPUOptions& options) { in GetOrCreate()
342 int GetPriority(int tf_gpu_id, const GPUOptions& options) { in GetPriority()
827 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()
955 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()
|
| D | gpu_device_test.cc | 78 GPUOptions* gpu_options = config->mutable_gpu_options();
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.h | 42 class GPUOptions; variable
99 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
153 EventMgr* GetEventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
|
| D | device_event_mgr.cc | 93 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
252 const GPUOptions& gpu_options) { in GetEventMgr()
|
| D | device_event_mgr_test.cc | 42 TEST_EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in TEST_EventMgr()
113 TEST_EventMgr em(stream_exec, GPUOptions()); in TEST()
122 TEST_EventMgr em(stream_exec, GPUOptions()); in TEST()
152 GPUOptions(), TfGpuId(0), memory_limit, /*peer_gpu_ids=*/{}); in GPUDeviceTestHelper()
447 TEST_EventMgr em(stream_exec, GPUOptions()); in BM_no_ops()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | virtual_gpu_test.py | 90 virtual_device_gpu_options = config_pb2.GPUOptions(
92 experimental=config_pb2.GPUOptions.Experimental(virtual_devices=[
93 config_pb2.GPUOptions.Experimental.VirtualDevices(
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | grpc_tensorflow_server.py | 108 config = config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(
|
| D | dist_session_debug_grpc_test.py | 89 gpu_options=config_pb2.GPUOptions(
|
| D | grpc_debug_test_server.py | 452 config = config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/python/ |
| D | modules_with_exports.py | 79 tf_export(v1=['GPUOptions'])(GPUOptions)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 738 gpu_options=config_pb2.GPUOptions(visible_device_list='0'))
759 gpu_options=config_pb2.GPUOptions(visible_device_list=''))
769 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count)))
777 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count-1)))
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 90 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST()
|
| D | convert_graph.cc | 643 GPUOptions gpu_options; in GetDeviceAndAllocator()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1077 config_pb2.GPUOptions.Experimental.VirtualDevices(
1089 return config_pb2.GPUOptions(
1092 experimental=config_pb2.GPUOptions.Experimental(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib_test.py | 185 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 84 gpu_options=config_pb2.GPUOptions(allow_growth=True))
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 328 gpu_options = config_pb2.GPUOptions()
|