| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_simple_allocator.h | 42 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
43 return sub_allocator_->GetMemoryType(); in GetMemoryType()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.h | 230 virtual AllocatorMemoryType GetMemoryType() const { in GetMemoryType() function
285 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
286 return wrapped_->GetMemoryType(); in GetMemoryType()
402 virtual AllocatorMemoryType GetMemoryType() const { in GetMemoryType() function
|
| D | cpu_allocator_impl.cc | 166 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function in tensorflow::__anonb46ab94d0111::CPUAllocator
208 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function in tensorflow::__anonb46ab94d0111::CPUAllocatorFactory::CPUSubAllocator
209 return cpu_allocator_->GetMemoryType(); in GetMemoryType()
|
| D | tracking_allocator.h | 71 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
72 return allocator_->GetMemoryType(); in GetMemoryType()
|
| D | tensor.h | 100 virtual AllocatorMemoryType GetMemoryType() const { in GetMemoryType() function
669 AllocatorMemoryType GetMemoryType() const { return buf_->GetMemoryType(); } in GetMemoryType() function
|
| D | tensor.cc | 123 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function in tensorflow::__anon2f56f8af0111::BufferBase
124 return alloc_->GetMemoryType(); in GetMemoryType()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | pool_allocator.h | 103 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
104 return allocator_->GetMemoryType(); in GetMemoryType()
169 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
|
| D | process_state.h | 151 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
152 return a_->GetMemoryType(); in GetMemoryType()
|
| D | bfc_allocator.h | 102 AllocatorMemoryType GetMemoryType() const override;
|
| D | bfc_allocator.cc | 1242 AllocatorMemoryType BFCAllocator::GetMemoryType() const { in GetMemoryType() function in tensorflow::BFCAllocator
1243 return sub_allocator_->GetMemoryType(); in GetMemoryType()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_debug_allocator.h | 77 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
78 return base_allocator_->GetMemoryType(); in GetMemoryType()
|
| D | gpu_cudamalloc_allocator.h | 39 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
|
| D | gpu_cudamallocasync_allocator.h | 96 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
|
| D | gpu_util.cc | 260 return tensor->GetMemoryType() == AllocatorMemoryType::kHostPageable; in NeedStaging()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_host_allocator.h | 64 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
|
| D | device_mem_allocator.h | 73 AllocatorMemoryType GetMemoryType() const override { in GetMemoryType() function
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | cupti_tracer.cc | 1472 static CUmemorytype GetMemoryType(CUdeviceptr ptr) { in GetMemoryType() function in tensorflow::profiler::__anon1bba5cd30111::CuptiDriverApiHookWithCudaEvent
1486 CUmemorytype src_type = GetMemoryType(src); in GetMemcpyType()
1487 CUmemorytype dst_type = GetMemoryType(dst); in GetMemcpyType()
|