Program Listing for File response_memory.hpp

Return to documentation for file (morpheus/_lib/include/morpheus/messages/memory/response_memory.hpp)

Copy
Copied!
            

#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

© Copyright 2023, NVIDIA. Last updated on Feb 3, 2023.