Searched refs:SYCLAllocator (Results 1 – 3 of 3) sorted by relevance
22 SYCLAllocator::SYCLAllocator(Eigen::QueueInterface* queue) in SYCLAllocator() function in tensorflow::SYCLAllocator30 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()
30 class SYCLAllocator : public Allocator {32 SYCLAllocator(Eigen::QueueInterface* queue);33 virtual ~SYCLAllocator() override;70 TF_DISALLOW_COPY_AND_ASSIGN(SYCLAllocator);
33 std::vector<SYCLAllocator*> m_sycl_allocator_; // owned118 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