Searched refs:buffer_usage_records (Results 1 – 2 of 2) sorted by relevance
486 std::vector<TensorUsageRecord<size_t>> buffer_usage_records; in AllocateMemoryForBuffers() local492 buffer_usage_records.size(); in AllocateMemoryForBuffers()493 buffer_usage_records.push_back({buffer_size, in AllocateMemoryForBuffers()500 buffer_usage_records, MemoryStrategy::GREEDY_BEST, &buffer_assignment)); in AllocateMemoryForBuffers()523 std::vector<bool> created_tensors(buffer_usage_records.size(), false); in AllocateMemoryForBuffers()524 shared_buffer_tensors_.resize(buffer_usage_records.size()); in AllocateMemoryForBuffers()
529 std::vector<TensorUsageRecord<size_t>> buffer_usage_records; in AllocateMemoryForBuffers() local539 buffer_usage_records.size(); in AllocateMemoryForBuffers()540 buffer_usage_records.push_back({buffer_size, in AllocateMemoryForBuffers()547 buffer_usage_records, MemoryStrategy::GREEDY_BEST, &buffer_assignment)); in AllocateMemoryForBuffers()555 std::vector<bool> created_tensors(buffer_usage_records.size(), false); in AllocateMemoryForBuffers()556 shared_buffer_tensors_.resize(buffer_usage_records.size()); in AllocateMemoryForBuffers()