Searched refs:GPUOptions (Results 1 – 25 of 30) sorted by relevance
12
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_event_mgr_test.cc | 101 EventMgr em(stream_exec, GPUOptions()); in TEST()
118 EventMgr em(stream_exec, GPUOptions()); in TEST()
151 EventMgr em(stream_exec, GPUOptions()); in TEST()
168 EventMgr em(stream_exec, GPUOptions()); in TEST()
187 EventMgr em(stream_exec, GPUOptions()); in TEST()
209 EventMgr em(stream_exec, GPUOptions()); in TEST()
231 EventMgr em(stream_exec, GPUOptions()); in TEST()
250 EventMgr em(stream_exec, GPUOptions()); in TEST()
|
| D | gpu_bfc_allocator.cc | 25 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue()
60 : GPUBFCAllocator(sub_allocator, total_memory, GPUOptions(), name) {} in GPUBFCAllocator()
64 const GPUOptions& gpu_options, in GPUBFCAllocator()
|
| D | gpu_bfc_allocator.h | 92 const GPUOptions& gpu_options, const string& name);
98 static bool GetAllowGrowthValue(const GPUOptions& gpu_options);
|
| D | gpu_event_mgr.h | 40 class GPUOptions; variable
66 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
|
| D | gpu_process_state.h | 83 virtual Allocator* GetGPUAllocator(const GPUOptions& options,
|
| D | gpu_event_mgr.cc | 92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
|
| D | gpu_bfc_allocator_test.cc | 234 GPUOptions options; in TEST()
522 GPUOptions options; in TestForceAllowGrowth()
|
| D | gpu_process_state.cc | 80 Allocator* GPUProcessState::GetGPUAllocator(const GPUOptions& options, in GetGPUAllocator()
|
| D | gpu_device_test.cc | 75 GPUOptions* gpu_options = config->mutable_gpu_options();
|
| D | gpu_device.cc | 240 const GPUOptions& options) { in GetOrCreate()
815 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()
887 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()
|
| /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/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/contrib/mpi_collectives/ |
| D | mpi_ops_test.py | 123 no_gpus = tf.GPUOptions(visible_device_list="")
128 one_gpu = tf.GPUOptions(visible_device_list=str(local_rank))
|
| D | mpi_allgather_test.py | 100 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
|
| D | mpi_allreduce_test.py | 105 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
|
| /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 | 87 gpu_options=config_pb2.GPUOptions(
|
| D | grpc_debug_test_server.py | 452 config = config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | plugin_test.py | 60 gpu_options = config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.50)
|
| /external/tensorflow/tensorflow/python/ |
| D | __init__.py | 161 tf_export(v1=['GPUOptions'])(GPUOptions)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 163 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST()
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | run_config.py | 308 gpu_options = config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 104 gpu_options=config_pb2.GPUOptions(allow_growth=True))
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 213 gpu_options = config_pb2.GPUOptions()
|
12