Program Listing for File inference_memory.hpp

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

Copy
Copied!
            

#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

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