Home
last modified time | relevance | path

Searched refs:SYCLAllocator (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_allocator.cc22 SYCLAllocator::SYCLAllocator(Eigen::QueueInterface* queue) in SYCLAllocator() function in tensorflow::SYCLAllocator
30 SYCLAllocator::~SYCLAllocator() { in ~SYCLAllocator()
36 string SYCLAllocator::Name() { return "device:SYCL"; } in Name()
38 void* SYCLAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()
59 void SYCLAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()
69 void SYCLAllocator::GetStats(AllocatorStats* stats) { in GetStats()
74 void SYCLAllocator::ClearStats() override { in ClearStats()
81 size_t SYCLAllocator::RequestedSize(void* ptr) { in RequestedSize()
Dsycl_allocator.h30 class SYCLAllocator : public Allocator {
32 SYCLAllocator(Eigen::QueueInterface* queue);
33 virtual ~SYCLAllocator() override;
70 TF_DISALLOW_COPY_AND_ASSIGN(SYCLAllocator);
Dsycl_device.h33 std::vector<SYCLAllocator*> m_sycl_allocator_; // owned
118 m_sycl_allocator_.push_back(new SYCLAllocator(m_queue_interface_.back())); in AddDevice()
138 SYCLAllocator* GetSYCLAllocator(size_t i = 0) const {
200 const string& physical_device_desc, SYCLAllocator* sycl_allocator, in SYCLDevice()
226 SYCLAllocator* sycl_allocator_; // not owned