Searched defs:custom_allocator (Results 1 – 3 of 3) sorted by relevance
48 std::shared_ptr<se::DeviceMemoryAllocator> custom_allocator() const { in custom_allocator() function
121 std::shared_ptr<se::DeviceMemoryAllocator> custom_allocator; in XlaPlatformInfoFromDevice() local
192 bool custom_allocator = (pAllocator != nullptr); in CreateObject() local249 bool custom_allocator = pAllocator != nullptr; in ValidateDestroyObject() local