| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ |
| H A D | device_event_mgr.cc | 93 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
252 const GPUOptions& gpu_options) { in GetEventMgr()
|
| H A D | device_event_mgr_test.cc | 42 TEST_EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in TEST_EventMgr()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_event_mgr.cc | 92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr()
251 const GPUOptions& gpu_options) { in GetEventMgr()
|
| H A 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
|
| H A D | gpu_device_test.cc | 81 GPUOptions* gpu_options = config->mutable_gpu_options(); local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | kernel_ops_pattern.cc | 219 const GpuBinaryOptions& gpu_options, in Emit()
253 GpuBinaryOptions gpu_options) { in Match()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_executable_run_options.cc | 58 const GpuExecutableRunOptions* gpu_options = in NcclExecuteParams() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt_device.cc | 117 GPUOptions gpu_options; in Acquire() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_compile_on_demand_op.cc | 82 xla::gpu::GpuExecutableRunOptions gpu_options; in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph_test.cc | 80 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
|
| H A D | convert_graph.cc | 654 GPUOptions gpu_options; in GetDeviceAndAllocator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_helpers.cc | 151 xla::gpu::GpuExecutableRunOptions& gpu_options) { in ResolveDeviceAssignment()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_execute_op.cc | 327 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | config.proto | 508 GPUOptions gpu_options = 6; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/ |
| H A D | xla_ops.cc | 354 xla::gpu::GpuExecutableRunOptions gpu_options; in ComputeAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | c_api_test_util.cc | 472 tensorflow::GPUOptions* gpu_options = in GetMultiClientServerDef() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_experimental.cc | 149 auto* gpu_options = config.mutable_gpu_options(); in TF_CreateConfig() local
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | config.pb.h | 6495 inline const ::tensorflow::GPUOptions& ConfigProto::gpu_options() const { in gpu_options() function
6500 ::tensorflow::GPUOptions* gpu_options) { in unsafe_arena_set_allocated_gpu_options()
6542 inline void ConfigProto::set_allocated_gpu_options(::tensorflow::GPUOptions* gpu_options) { in set_allocated_gpu_options()
|
| D | config.pb.cc | 4252 ConfigProto::_Internal::gpu_options(const ConfigProto* msg) { in gpu_options() function in tensorflow::ConfigProto::_Internal
|