Program Listing for File multi_response_probs.hpp

            

            
#pragma once

#include "morpheus/messages/memory/response_memory_probs.hpp"
#include "morpheus/messages/meta.hpp"
#include "morpheus/messages/multi.hpp"
#include "morpheus/messages/multi_response.hpp"
#include "morpheus/objects/tensor_object.hpp"

#include <cudf/types.hpp>
#include <pybind11/pytypes.h>

#include <cstddef>// for size_t
#include <memory>

namespace morpheus {
/****** Component public implementations *******************/
/****** MultiResponseProbsMessage****************************************/

#pragma GCC visibility push(default)
class MultiResponseProbsMessage : public DerivedMultiMessage<MultiResponseProbsMessage, MultiResponseMessage>
{
  public:
    MultiResponseProbsMessage(const MultiResponseProbsMessage &other) = default;

    MultiResponseProbsMessage(std::shared_ptr<morpheus::MessageMeta> meta,
                              size_t mess_offset,
                              size_t mess_count,
                              std::shared_ptr<morpheus::ResponseMemoryProbs> memory,
                              size_t offset,
                              size_t count);

    const TensorObject get_probs() const;

    void set_probs(const TensorObject &probs);
};

/****** MultiResponseProbsMessageInterfaceProxy *************************/
struct MultiResponseProbsMessageInterfaceProxy
{
    static std::shared_ptr<MultiResponseProbsMessage> init(std::shared_ptr<MessageMeta> meta,
                                                           cudf::size_type mess_offset,
                                                           cudf::size_type mess_count,
                                                           std::shared_ptr<ResponseMemoryProbs> memory,
                                                           cudf::size_type offset,
                                                           cudf::size_type count);

    static std::shared_ptr<morpheus::ResponseMemoryProbs> memory(MultiResponseProbsMessage &self);

    static std::size_t offset(MultiResponseProbsMessage &self);

    static std::size_t count(MultiResponseProbsMessage &self);

    static pybind11::object probs(MultiResponseProbsMessage &self);
};
#pragma GCC visibility pop// end of group
}  // namespace morpheus