Skip to content
Snippets Groups Projects

Feat/operator global average pooling

Compare and
13 files
+ 469
156
Compare changes
  • Side-by-side
  • Inline
Files
13
@@ -44,10 +44,10 @@ private:
@@ -44,10 +44,10 @@ private:
/// @brief Set of nodes included in the graphview with names
/// @brief Set of nodes included in the graphview with names
std::map<std::string, NodePtr> mNodeRegistry;
std::map<std::string, NodePtr> mNodeRegistry;
/// @brief GraphView inputs
/// @brief GraphView inputs IOIndex_t designates the input number
std::vector<std::pair<NodePtr, IOIndex_t>> mInputNodes;
std::vector<std::pair<NodePtr, IOIndex_t>> mInputNodes;
/// @brief GraphView outputs
/// @brief GraphView outputs IOIndex_t designates the input number
std::vector<std::pair<NodePtr, IOIndex_t>> mOutputNodes;
std::vector<std::pair<NodePtr, IOIndex_t>> mOutputNodes;
public:
public:
Loading