Program Listing for File tensor_memory.cpp

            

            
#include "morpheus/messages/memory/tensor_memory.hpp"

#include <string>
#include <vector>

namespace morpheus {
/****** Component public implementations *******************/
/****** TensorMemory****************************************/
TensorMemory::TensorMemory(size_t count) : count(count) {}
TensorMemory::TensorMemory(size_t count, tensor_map_t &&tensors) : count(count), tensors(std::move(tensors)) {}

bool TensorMemory::has_tensor(const std::string &name) const
{
    return this->tensors.find(name) != this->tensors.end();
}

TensorMemory::tensor_map_t TensorMemory::copy_tensor_ranges(
    const std::vector<std::pair<TensorIndex, TensorIndex>> &ranges, size_t num_selected_rows) const
{
    tensor_map_t tensors;
    for (const auto &p : this->tensors)
    {
        tensors.insert(std::pair{p.first, p.second.copy_rows(ranges, num_selected_rows)});
    }

    return tensors;
}

}  // namespace morpheus