Searched refs:xla_allocator (Results 1 – 4 of 4) sorted by relevance
42 std::unique_ptr<XlaAllocator> xla_allocator, in XlaPlatformInfo() argument47 xla_allocator_(std::move(xla_allocator)), in XlaPlatformInfo()
64 std::unique_ptr<XlaAllocator> xla_allocator; in PlatformInfoFromContext() local96 xla_allocator = absl::make_unique<XlaAllocator>( in PlatformInfoFromContext()101 std::move(xla_allocator), device_allocator); in PlatformInfoFromContext()
145 xla::DeviceMemoryAllocator* xla_allocator,
195 xla::LocalClient* client, xla::DeviceMemoryAllocator* xla_allocator, in XlaComputationLaunchContext() argument198 xla_allocator_(xla_allocator), in XlaComputationLaunchContext()