Searched defs:device_allocator (Results 1 – 13 of 13) sorted by relevance
107 se::DeviceMemoryAllocator* device_allocator() const { in device_allocator() function110 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator()230 se::DeviceMemoryAllocator* device_allocator = nullptr; member248 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()275 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()300 se::DeviceMemoryAllocator* device_allocator) { in Compile()
78 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()110 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
73 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()121 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
338 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule()724 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
35 std::shared_ptr<se::DeviceMemoryAllocator> device_allocator) in XlaPlatformInfo()
48 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()54 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
108 Allocator* device_allocator; member
119 Allocator* device_allocator = nullptr; in GetPluggableDeviceAllocator() local
249 Allocator* device_allocator = process_state->GetPluggableDeviceAllocator( in CreatePluggableDevice() local
143 const std::string& physical_device_desc, Allocator* device_allocator, in PluggableDevice()
31 se::DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
194 std::shared_ptr<se::DeviceMemoryAllocator> device_allocator; member
697 std::pair<int, Allocator*> device_allocator = in CreateStaticEngine() local