Home
last modified time | relevance | path

Searched refs:use_unified_memory (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_process_state.cc111 bool use_unified_memory = options.per_process_gpu_memory_fraction() > 1.0 || in GetPluggableDeviceAllocator() local
112 options.experimental().use_unified_memory(); in GetPluggableDeviceAllocator()
116 platform_device_id, use_unified_memory, in GetPluggableDeviceAllocator()
H A Dpluggable_device_factory.cc98 device_options.experimental().use_unified_memory()) { in SingleVirtualDeviceMemoryLimit()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/
H A Ddevice_mem_allocator.h33 bool use_unified_memory, in DeviceMemAllocator() argument
39 use_unified_memory_(use_unified_memory) { in DeviceMemAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_process_state.cc112 options.experimental().use_unified_memory()) { in CreateSubAllocator()
146 options.experimental().use_unified_memory()), in CreateSubAllocator()
H A Dgpu_device.cc1104 gpu_options.experimental().use_unified_memory()) && in SingleVirtualDeviceMemoryLimit()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-g-p-u-options.pbtxt70 name: "use_unified_memory"
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto181 bool use_unified_memory = 2; field
/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/
Dconfig.pb.h605 bool use_unified_memory() const; in Swap()
4411 inline bool GPUOptions_Experimental::use_unified_memory() const { in use_unified_memory() function