| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | temporary_memory_manager.cc | 31 DeviceMemoryBase device_memory = it->first; in ForceDeallocateAll() local
37 const DeviceMemoryBase& device_memory, uint64_t generation, in MarkFinalized()
56 DeviceMemoryBase device_memory = it->first; in DeallocateFinalizedTemporaries() local
67 bool TemporaryMemoryManager::IsFinalized(const DeviceMemoryBase& device_memory, in IsFinalized()
83 bool TemporaryMemoryManager::HasAllocated(const DeviceMemoryBase& device_memory, in HasAllocated()
97 DeviceMemoryBase device_memory = in AllocateArrayBase() local
|
| H A D | temporary_device_memory.cc | 34 const DeviceMemoryBase& TemporaryDeviceMemoryBase::device_memory() const { in device_memory() function in stream_executor::TemporaryDeviceMemoryBase
58 Stream* parent, DeviceMemoryBase device_memory, in TemporaryDeviceMemoryBase()
|
| H A D | temporary_device_memory.h | 119 const DeviceMemory<T>& device_memory() const { in device_memory() function
|
| /aosp_15_r20/external/rust/android-crates-io/crates/vulkano/src/memory/allocator/ |
| D | suballocator.rs | 312 let device_memory = self.device_memory(); in create_memory_range() localVariable
351 pub fn device_memory(&self) -> &DeviceMemory { in device_memory() method
962 device_memory: Arc<DeviceMemory>, field
988 let device_memory = region in new() localVariable
1544 device_memory: Arc<DeviceMemory>, field
1584 let device_memory = region in new() localVariable
2084 device_memory: Arc<DeviceMemory>, field
2097 let device_memory = region in new() localVariable
2249 device_memory: Arc<DeviceMemory>, field
2272 let device_memory = region in new() localVariable
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tracked_device_buffer.h | 173 absl::InlinedVector<se::DeviceMemoryBase, 1>& device_memory() { in device_memory() function
176 const absl::InlinedVector<se::DeviceMemoryBase, 1>& device_memory() const { in device_memory() function
|
| H A D | tracked_device_buffer.cc | 167 absl::Span<se::DeviceMemoryBase const> device_memory, in TrackedDeviceBuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | transfer_manager.cc | 296 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in WriteTupleIndexTablesAsync() local
322 se::DeviceMemoryBase device_memory = device_buffer.buffer({}); in WriteRootTupleIndexTable() local
341 se::DeviceMemoryBase device_memory = in WriteRootTupleIndexTable() local
|
| H A D | allocation_tracker.cc | 208 se::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount()
221 Status AllocationTracker::DecrementRefCount(se::DeviceMemoryBase device_memory, in DecrementRefCount()
|
| H A D | allocation_tracker.h | 77 se::OwningDeviceMemory device_memory; member
|
| H A D | generic_transfer_manager.cc | 115 se::DeviceMemoryBase device_memory = device_buffer.buffer(index); in TransferLiteralToDeviceAsync() local
|
| /aosp_15_r20/external/crosvm/rutabaga_gfx/src/rutabaga_gralloc/ |
| H A D | vulkano_gralloc.rs | 477 let device_memory = DeviceMemory::allocate( in allocate_memory() localVariable
508 let device_memory = unsafe { in import_and_map() localVariable
|
| /aosp_15_r20/external/perfetto/protos/perfetto/trace/gpu/ |
| H A D | vulkan_memory_event.proto | 99 optional fixed64 device_memory = 17; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | tpu_executable_interface.cc | 129 MaybeOwningDeviceMemory* device_memory = in AllocateOutputMemoryWithInputReuse() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | executable_base.cc | 199 MaybeOwningDeviceMemory* device_memory = in AllocateOutputMemoryWithInputReuse() local
|
| /aosp_15_r20/external/virglrenderer/src/venus/ |
| H A D | vkr_allocator.c | 47 VkDeviceMemory device_memory; member
|
| H A D | vkr_common.h | 116 VkDeviceMemory device_memory; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | buffer_donation_test.cc | 115 [&](const ShapeIndex& index, MaybeOwningDeviceMemory* device_memory) { in RunAndCheck()
|
| /aosp_15_r20/external/rust/android-crates-io/crates/vulkano/src/memory/ |
| D | mod.rs | 115 mod device_memory; module
|
| /aosp_15_r20/external/mesa3d/src/freedreno/vulkan/ |
| H A D | tu_rmv.cc | 159 struct tu_device_memory *device_memory) in tu_rmv_log_heap_create()
|
| /aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_gpu_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace/gpu/ |
| D | vulkan_memory_event.pbzero.h | 188 uint64_t device_memory() const { return at<17>().as_uint64(); } in device_memory() function
|
| /aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include_gen/external/perfetto/perfetto_protos_perfetto_trace_gpu_zero_gen_headers/gen/external/perfetto/protos/perfetto/trace/gpu/ |
| H A D | vulkan_memory_event.pbzero.h | 188 uint64_t device_memory() const { return at<17>().as_uint64(); } in device_memory() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_blas.cc | 664 DeviceMemory<MAPPED_T> *device_memory, in ReorganizeMemory()
716 DeviceMemory<typename RocBlasTypeConversionHelper<T>::mapped_type> in AllocateStridedBuffer()
|
| /aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_protos_perfetto_trace_gpu_lite_gen_headers/gen/external/perfetto/protos/perfetto/trace/gpu/ |
| D | vulkan_memory_event.pb.h | 1268 inline ::uint64_t VulkanMemoryEvent::device_memory() const { in device_memory() function
|
| /aosp_15_r20/prebuilts/runtime/mainline/platform/sdk/include_gen/external/perfetto/perfetto_protos_perfetto_trace_gpu_lite_gen_headers/gen/external/perfetto/protos/perfetto/trace/gpu/ |
| H A D | vulkan_memory_event.pb.h | 1268 inline ::uint64_t VulkanMemoryEvent::device_memory() const { in device_memory() function
|
| /aosp_15_r20/out/soong/.intermediates/external/perfetto/perfetto_src_trace_processor_tables_tables_python/gen/src/trace_processor/tables/ |
| D | profiler_tables_py.h | 8663 static constexpr uint32_t device_memory = 8; member
8681 using device_memory = TypedColumn<std::optional<int64_t>>; member
8729 std::optional<int64_t> device_memory; member
8815 ColumnType::device_memory::type device_memory() const { in device_memory() function
8956 ColumnType::device_memory::type device_memory() const { in device_memory() function
9405 const TypedColumn<std::optional<int64_t>>& device_memory() const { in device_memory() function
|