Program Listing for File 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