Searched defs:device_allocator (Results 1 – 14 of 14) sorted by relevance
31 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()43 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()56 se::DeviceMemoryAllocator* device_allocator) { in Compile()
143 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()458 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()562 se::DeviceMemoryAllocator* device_allocator) { in Compile()
71 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()88 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
121 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule()319 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()335 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()
114 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()156 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
81 se::DeviceMemoryAllocator* device_allocator() const { in device_allocator() function84 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator()
27 se::DeviceMemoryAllocator* device_allocator) { in Compile()
366 se::DeviceMemoryAllocator* device_allocator) { in BuildExecutables()806 se::StreamExecutor* executor, se::DeviceMemoryAllocator* device_allocator) { in BuildExecutable()
30 se::DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
46 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()52 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
43 se::DeviceMemoryAllocator* device_allocator) in XlaPlatformInfo()
127 se::DeviceMemoryAllocator* device_allocator) { in Compile()
112 pub fn device_allocator(&mut self) -> &mut AddressAllocator { in device_allocator() method
344 se::DeviceMemoryAllocator* device_allocator = nullptr; member