Searched defs:allocated_buffers_ (Results 1 – 4 of 4) sorted by relevance
54 std::vector<se::OwningDeviceMemory> allocated_buffers_; variable
77 std::vector<se::OwningDeviceMemory> allocated_buffers_; member in xla::gpu::__anon559be1860111::ScratchAllocator
124 std::vector<std::pair<OwningDeviceMemory, int64>> allocated_buffers_; variable
225 absl::flat_hash_set<const HloValue*> allocated_buffers_; variable