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