Searched refs:shared_buffers_ (Results 1 – 4 of 4) sorted by relevance
549 shared_buffers_.resize(buffer_assignment.object_sizes.size()); in AllocateMemoryForBuffers()552 context, &shared_buffers_[i])); in AllocateMemoryForBuffers()566 *context, shared_buffers_[buffer_index].GetMemoryPtr(), shape, in AllocateMemoryForBuffers()690 for (const auto& b : shared_buffers_) { in GetSizeOfMemoryAllocatedForIntermediateTensors()
224 std::vector<Buffer> shared_buffers_; variable
504 shared_buffers_.resize(buffer_assignment.object_sizes.size()); in AllocateMemoryForBuffers()518 shared_buffers_[i] = in AllocateMemoryForBuffers()536 shared_buffers_[buffer_index], tensor_dummy.shape, in AllocateMemoryForBuffers()
217 std::vector<id<MTLBuffer>> shared_buffers_; variable