1234567891011121314151617181920212223242526272829303132 |
- #include "nodes/base/base.hpp"
- #include "nodes/infer/inferNode.hpp"
- #include <unordered_map>
- namespace Node
- {
- void InferNode::work()
- {
- printf("InferNode %s\n", name_.c_str());
- while (running_)
- {
-
- for (auto& input_buffer : input_buffers_)
- {
- std::shared_ptr<meta::MetaData> metaData;
- if (!input_buffer.second->try_pop(metaData))
- {
- continue;
- }
- printf("Node %s get data from %s\n", name_.c_str(), input_buffer.first.c_str());
- // do something
- for (auto& output_buffer : output_buffers_)
- {
- printf("Node %s push data to %s\n", name_.c_str(), output_buffer.first.c_str());
- output_buffer.second->push(metaData);
- }
- }
- }
- };
- } // namespace Node
|