Searched defs:allocated_buffers_ (Results 1 – 3 of 3) sorted by relevance
126 std::vector<std::pair<OwningDeviceMemory, int64_t>> allocated_buffers_; variable
231 absl::flat_hash_set<const HloValue*> allocated_buffers_; variable
84 std::vector<se::OwningDeviceMemory> allocated_buffers_; member in xla::gpu::__anon8ba6008b0111::ScratchAllocator