Searched refs:xla_allocator (Results 1 – 2 of 2) sorted by relevance
136 se::DeviceMemoryAllocator* xla_allocator,
190 xla::LocalClient* client, se::DeviceMemoryAllocator* xla_allocator, in XlaComputationLaunchContext() argument193 xla_allocator_(xla_allocator), in XlaComputationLaunchContext()