Home
last modified time | relevance | path

Searched defs:GetMemoryType (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dallocator.h230 virtual AllocatorMemoryType GetMemoryType() const { in GetMemoryType() function
285 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
402 virtual AllocatorMemoryType GetMemoryType() const { in GetMemoryType() function
H A Dcpu_allocator_impl.cc166 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function in tensorflow::__anon2523aecd0111::CPUAllocator
208 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function in tensorflow::__anon2523aecd0111::CPUAllocatorFactory::CPUSubAllocator
H A Dtensor.h100 virtual AllocatorMemoryType GetMemoryType() const { in GetMemoryType() function
669 AllocatorMemoryType GetMemoryType() const { return buf_->GetMemoryType(); } in GetMemoryType() function
H A Dtracking_allocator.h71 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
H A Dtensor.cc123 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function in tensorflow::__anona4d3ae2f0111::BufferBase
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dpool_allocator.h103 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
169 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
H A Dprocess_state.h151 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
H A Dbfc_allocator.cc1242 AllocatorMemoryType BFCAllocator::GetMemoryType() const { in GetMemoryType() function in tensorflow::BFCAllocator
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_cudamalloc_allocator.h39 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
H A Dgpu_debug_allocator.h77 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
H A Dgpu_cudamallocasync_allocator.h96 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_simple_allocator.h42 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/
H A Ddevice_host_allocator.h64 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
H A Ddevice_mem_allocator.h73 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
/aosp_15_r20/hardware/google/gfxstream/host/vulkan/
DVkDecoderSnapshotUtils.cpp24 uint32_t GetMemoryType(const PhysicalDeviceInfo& physicalDevice, in GetMemoryType() function
/aosp_15_r20/hardware/google/gfxstream/common/detector/
DVulkan.cpp48 uint32_t GetMemoryType(const vkhpp::PhysicalDevice& physical_device, in GetMemoryType() function
/aosp_15_r20/device/google/cuttlefish/tests/graphics/vulkan/
Dsample_base.cpp45 Result<uint32_t> GetMemoryType(const vkhpp::PhysicalDevice& physical_device, in GetMemoryType() function
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/
H A Dcupti_tracer.cc1472 static CUmemorytype GetMemoryType(CUdeviceptr ptr) { in GetMemoryType() function in tensorflow::profiler::__anon2fb572530111::CuptiDriverApiHookWithCudaEvent