Skip to content
Snippets Groups Projects
Commit 07350033 authored by Octave Perrin's avatar Octave Perrin
Browse files

continuation

parent b0d95184
No related branches found
No related tags found
No related merge requests found
......@@ -182,10 +182,16 @@ public:
/** @brief Get reference to the set of output Nodes. */
std::set<NodePtr> outputNodes() const;
/** @brief Assess if the given Node is an input Node of the GraphView object. */
/** @brief Assess if the given Node is an input Node of the GraphView object.
* @param Node The tested Node
* @return bool True if the Node is an input of the GraphView, False otherwise
*/
bool isInputNode(const NodePtr& nodePtr) const;
/** @brief Assess if the given Node is an output Node of the GraphView object. */
/** @brief Assess if the given Node is an output Node of the GraphView object.
* @param Node The tested Node
* @return bool True if the Node is an output of the GraphView, False otherwise
*/
bool isOutputNode(const NodePtr& nodePtr) const;
/**
......@@ -228,6 +234,7 @@ public:
* @return const std::vector<std::pair<NodePtr, IOIndex_t>>&
*/
inline const std::vector<std::pair<NodePtr, IOIndex_t>>& getOrderedInputs() const noexcept { return mInputNodes; };
/**
* @brief Get outputs of the current GraphView with their associated id.
* The rank of the nodes are their rank in the vector.
......@@ -281,8 +288,8 @@ public:
* @param nodeName Name of the Node of which to show the output.
* @return std::vector<std::vector<std::pair<NodePtr, IOIndex_t>>>
*/
std::vector<std::vector<std::pair<NodePtr, IOIndex_t>>> outputs(
const std::string& nodeName) const;
std::vector<std::vector<std::pair<NodePtr, IOIndex_t>>> outputs(const std::string& nodeName) const;
/**
* @brief Assert Datatype, Backend, data format and dimensions along the GraphView are coherent.
* If not, apply the required transformations.
......@@ -323,10 +330,12 @@ public:
* @param DeviceIdx_t: device Backend device to be set
*/
void setBackend(const std::string& backend, const DeviceIdx_t device = 0) const;
/** @brief Set the same data type for each Operator of the GraphView object's Nodes.
* @param DataType: datatype DataType to be set
*/
void setDataType(const DataType& datatype) const;
/** @brief Set the same data format for each Operator of the GraphView object's Nodes.
* @param DataFormat: dataformat DataFormat to be set
*/
......@@ -342,6 +351,7 @@ public:
* @return std::set<NodePtr>
*/
std::set<NodePtr> getParents() const;
/**
* @brief Get parents Nodes of the specified Node.
* @param nodeName Name of the Node.
......@@ -372,7 +382,6 @@ public:
*/
std::vector<std::vector<NodePtr>> getChildren(const std::string nodeName) const;
/**
* @brief Get children Nodes of the specified Node.
* @param otherNode The Node
......@@ -703,6 +712,7 @@ private:
* @return GraphView GraphView containing all nodes with a path to node.
*/
std::shared_ptr<GraphView> getConnectedGraphView(std::shared_ptr<Node> node);
} // namespace Aidge
#endif /* AIDGE_CORE_GRAPH_GRAPHVIEW_H_ */
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment