| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_bfc_allocator.cc | 23 const GPUOptions& gpu_options, bool force_memory_growth_requested) { in GetAllowGrowthValue()
97 const GPUOptions& gpu_options, const string& name, in PluggableDeviceBFCAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| 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_device.cc | 932 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings()
1085 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit()
1286 const auto& gpu_options = options.config.gpu_options(); in CreateDevices() local
|
| D | gpu_device_test.cc | 81 GPUOptions* gpu_options = config->mutable_gpu_options(); local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable_run_options.cc | 58 const GpuExecutableRunOptions* gpu_options = in NcclExecuteParams() local
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_device.cc | 117 GPUOptions gpu_options; in Acquire() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 82 xla::gpu::GpuExecutableRunOptions gpu_options; in Run() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 80 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
|
| D | convert_graph.cc | 654 GPUOptions gpu_options; in GetDeviceAndAllocator() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | kernel_ops_pattern.cc | 219 const GpuBinaryOptions& gpu_options, in Emit()
253 GpuBinaryOptions gpu_options) { in Match()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 151 xla::gpu::GpuExecutableRunOptions& gpu_options) { in ResolveDeviceAssignment()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 327 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 508 GPUOptions gpu_options = 6; field
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 354 xla::gpu::GpuExecutableRunOptions gpu_options; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 472 tensorflow::GPUOptions* gpu_options = in GetMultiClientServerDef() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 149 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig() local
|