Program Listing for File multi_response_probs.hpp

Return to documentation for file (morpheus/_lib/include/morpheus/messages/multi_response_probs.hpp)

Copy
Copied!
            

#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

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.