↰ Return to documentation for file (morpheus/_lib/include/morpheus/messages/memory/inference_memory.hpp
)
#pragma once
#include "morpheus/messages/memory/tensor_memory.hpp"
#include <cstddef>
#include <string>
namespace morpheus {
#pragma GCC visibility push(default)
class InferenceMemory : public TensorMemory
{
public:
InferenceMemory(size_t count);
InferenceMemory(size_t count, tensor_map_t&& tensors);
bool has_input(const std::string& name) const;
};
/****** InferenceMemoryInterfaceProxy *************************/
struct InferenceMemoryInterfaceProxy
{
static std::size_t get_count(InferenceMemory& self);
};
#pragma GCC visibility pop
// end of group
} // namespace morpheus