↰ Return to documentation for file (morpheus/_lib/src/messages/multi_inference_fil.cpp
)
#include "morpheus/messages/multi_inference_fil.hpp"
#include "morpheus/messages/memory/inference_memory.hpp"
#include "morpheus/messages/meta.hpp"
#include "morpheus/messages/multi_inference.hpp"
#include <cudf/types.hpp>
#include <memory>
#include <utility>
namespace morpheus {
/****** Component public implementations *******************/
/****** MultiInferenceFILMessage****************************************/
MultiInferenceFILMessage::MultiInferenceFILMessage(std::shared_ptr<morpheus::MessageMeta> meta,
size_t mess_offset,
size_t mess_count,
std::shared_ptr<morpheus::InferenceMemory> memory,
size_t offset,
size_t count) :
MultiInferenceMessage(meta, mess_offset, mess_count, memory, offset, count)
{}
const TensorObject MultiInferenceFILMessage::get_input__0() const
{
return this->get_input("input__0");
}
void MultiInferenceFILMessage::set_input__0(const TensorObject &input__0)
{
this->set_input("input__0", input__0);
}
const TensorObject MultiInferenceFILMessage::get_seq_ids() const
{
return this->get_input("seq_ids");
}
void MultiInferenceFILMessage::set_seq_ids(const TensorObject &seq_ids)
{
this->set_input("seq_ids", seq_ids);
}
/****** MultiInferenceFILMessageInterfaceProxy *************************/
std::shared_ptr<MultiInferenceFILMessage> MultiInferenceFILMessageInterfaceProxy::init(
std::shared_ptr<MessageMeta> meta,
cudf::size_type mess_offset,
cudf::size_type mess_count,
std::shared_ptr<InferenceMemory> memory,
cudf::size_type offset,
cudf::size_type count)
{
return std::make_shared<MultiInferenceFILMessage>(
std::move(meta), mess_offset, mess_count, std::move(memory), offset, count);
}
std::shared_ptr<morpheus::InferenceMemory> MultiInferenceFILMessageInterfaceProxy::memory(
MultiInferenceFILMessage &self)
{
DCHECK(std::dynamic_pointer_cast<morpheus::InferenceMemory>(self.memory) != nullptr);
return std::static_pointer_cast<morpheus::InferenceMemory>(self.memory);
}
std::size_t MultiInferenceFILMessageInterfaceProxy::offset(MultiInferenceFILMessage &self)
{
return self.offset;
}
std::size_t MultiInferenceFILMessageInterfaceProxy::count(MultiInferenceFILMessage &self)
{
return self.count;
}
} // namespace morpheus
// Created by drobison on 3/17/22.
//