Searched refs:XlaDeviceAllocator (Results 1 – 3 of 3) sorted by relevance
33 XlaDeviceAllocator::XlaDeviceAllocator( in XlaDeviceAllocator() function in tensorflow::XlaDeviceAllocator37 XlaDeviceAllocator::~XlaDeviceAllocator() = default;39 string XlaDeviceAllocator::Name() { return "xla"; } in Name()41 void* XlaDeviceAllocator::AllocateRaw(size_t alignment, size_t num_bytes) { in AllocateRaw()51 void XlaDeviceAllocator::DeallocateRaw(void* ptr) { in DeallocateRaw()55 absl::optional<AllocatorStats> XlaDeviceAllocator::GetStats() { in GetStats()
35 class XlaDeviceAllocator : public Allocator {37 XlaDeviceAllocator(se::StreamExecutor* stream_executor);38 ~XlaDeviceAllocator() override;
68 static XlaDeviceAllocator* GetOrCreateXlaDeviceAllocator(79 std::unique_ptr<XlaDeviceAllocator>,94 XlaDeviceAllocator* XlaDeviceAllocatorState::GetOrCreateXlaDeviceAllocator( in GetOrCreateXlaDeviceAllocator()104 std::unique_ptr<XlaDeviceAllocator> alloc = in GetOrCreateXlaDeviceAllocator()105 absl::make_unique<XlaDeviceAllocator>( in GetOrCreateXlaDeviceAllocator()107 XlaDeviceAllocator* alloc_ptr = alloc.get(); in GetOrCreateXlaDeviceAllocator()