Program Listing for File response_memory_probs.hpp

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

Copy
Copied!
            

#pragma once #include "morpheus/messages/memory/response_memory.hpp" #include "morpheus/messages/memory/tensor_memory.hpp" #include "morpheus/objects/tensor_object.hpp" #include <cudf/types.hpp> #include <pybind11/pytypes.h> #include <cstddef> #include <memory> namespace morpheus { /****** Component public implementations *******************/ /****** ResponseMemoryProbs*********************************/ class ResponseMemoryProbs : public ResponseMemory { public: ResponseMemoryProbs(size_t count, TensorObject probs); ResponseMemoryProbs(size_t count, tensor_map_t &&tensors); const TensorObject &get_probs() const; void set_probs(TensorObject probs); }; /****** ResponseMemoryProbsInterfaceProxy*******************/ #pragma GCC visibility push(default) struct ResponseMemoryProbsInterfaceProxy { static std::shared_ptr<ResponseMemoryProbs> init(cudf::size_type count, pybind11::object probs); static std::size_t count(ResponseMemoryProbs &self); static pybind11::object get_probs(ResponseMemoryProbs &self); static void set_probs(ResponseMemoryProbs &self, pybind11::object cupy_values); }; #pragma GCC visibility pop // end of group } // namespace morpheus

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