↰ Return to documentation for file (morpheus/_lib/include/morpheus/messages/memory/response_memory.hpp
)
#pragma once
#include "morpheus/messages/memory/tensor_memory.hpp"
#include "morpheus/objects/tensor_object.hpp"// for TensorObject
#include <pybind11/pytypes.h>
#include <cstddef>// for size_t
#include <string>
namespace morpheus {
/****** Component public implementations *******************/
/****** ResponseMemory****************************************/
class ResponseMemory : public TensorMemory
{
public:
ResponseMemory(size_t count);
ResponseMemory(size_t count, tensor_map_t &&tensors);
bool has_output(const std::string &name) const;
};
/****** ResponseMemoryInterfaceProxy *************************/
#pragma GCC visibility push(default)
struct ResponseMemoryInterfaceProxy
{
static pybind11::object get_output(ResponseMemory &self, const std::string &name);
static TensorObject get_output_tensor(ResponseMemory &self, const std::string &name);
};
#pragma GCC visibility pop
// end of group
} // namespace morpheus