Searched defs:device_allocator (Results 1 – 13 of 13) sorted by relevance
85 se::DeviceMemoryAllocator* device_allocator() const { in device_allocator() function88 void set_device_allocator(se::DeviceMemoryAllocator* device_allocator) { in set_device_allocator()168 se::DeviceMemoryAllocator* device_allocator = nullptr; member186 se::DeviceMemoryAllocator* device_allocator) { in RunHloPasses()215 se::DeviceMemoryAllocator* device_allocator) { in RunBackend()233 se::DeviceMemoryAllocator* device_allocator) { in Compile()
35 std::shared_ptr<se::DeviceMemoryAllocator> device_allocator) in XlaPlatformInfo()
46 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()52 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
108 Allocator* device_allocator; member
119 Allocator* device_allocator = nullptr; in GetPluggableDeviceAllocator() local
248 Allocator* device_allocator = process_state->GetPluggableDeviceAllocator( in CreatePluggableDevice() local
143 const std::string& physical_device_desc, Allocator* device_allocator, in PluggableDevice()
111 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()162 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
76 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloConvolutionCanonicalization()
242 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloModule()538 se::DeviceMemoryAllocator* device_allocator) { in OptimizeHloPostLayoutAssignment()
31 se::DeviceMemoryAllocator* ExecutableBuildOptions::device_allocator() const { in device_allocator() function in xla::ExecutableBuildOptions
198 std::shared_ptr<se::DeviceMemoryAllocator> device_allocator; member
433 std::pair<int, Allocator*> device_allocator = in CreateTRTNode() local