Searched refs:XlaAllocator (Results 1 – 4 of 4) sorted by relevance
36 class XlaAllocator; variable111 class XlaAllocator : public xla::DeviceMemoryAllocator {113 XlaAllocator(const se::Platform* platform, Allocator* wrapped);114 ~XlaAllocator() override;
168 XlaAllocator::XlaAllocator(const se::Platform* platform, Allocator* wrapped) in XlaAllocator() function in tensorflow::XlaAllocator171 XlaAllocator::~XlaAllocator() {} in ~XlaAllocator()173 xla::StatusOr<xla::OwningDeviceMemory> XlaAllocator::Allocate( in Allocate()189 Status XlaAllocator::Deallocate(int device_ordinal, se::DeviceMemoryBase mem) { in Deallocate()
42 std::unique_ptr<XlaAllocator> xla_allocator, in XlaPlatformInfo()88 std::unique_ptr<XlaAllocator> xla_allocator_;
64 std::unique_ptr<XlaAllocator> xla_allocator; in PlatformInfoFromContext()96 xla_allocator = absl::make_unique<XlaAllocator>( in PlatformInfoFromContext()