↰ Return to documentation for file (morpheus/_lib/include/morpheus/messages/memory/response_memory_probs.hpp
)
#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