Searched defs:device_allocator (Results 1 – 9 of 9) sorted by relevance
86 se::DeviceMemoryAllocator* device_allocator() const { in device_allocator() function89 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator()169 se::DeviceMemoryAllocator* device_allocator = nullptr; member187 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()216 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()234 se::DeviceMemoryAllocator* device_allocator) { in Compile()
35 se::DeviceMemoryAllocator* device_allocator) in XlaPlatformInfo()
46 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()52 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
75 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()
109 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()162 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
151 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule()404 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
31 se::DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
195 se::DeviceMemoryAllocator* device_allocator = nullptr; member
434 std::pair<int, Allocator*> device_allocator = in CreateTRTNode() local