Searched refs:buffer_set (Results 1 – 4 of 4) sorted by relevance
44 buffer_set: &'a mut dyn BufferSet, field60 pub fn new<D: BufferSet>(requested_frames: usize, buffer_set: &'a mut D) -> Self { in new()63 buffer_set, in new()100 self.buffer_set.callback(offset, frames) in set_buffer_offset_and_frames()109 self.buffer_set.ignore() in ignore_request()
65 let buffer_set = (0..num_listios).map(|_| { in test_lio_listio_resubmit() localVariable77 &buffer_set[i][j][..], in test_lio_listio_resubmit()
327 absl::flat_hash_set<PjRtBuffer*> buffer_set; in HeapProfile() local333 if (!buffer->is_deleted() && buffer_set.insert(buffer->buffer()).second) { in HeapProfile()
137 auto& buffer_set = in GetBufferSet() local143 [&buffer_set](const ShapeIndex& /*index*/, in GetBufferSet()145 buffer_set->insert(buffers.begin(), buffers.end()); in GetBufferSet()147 return buffer_set.get(); in GetBufferSet()