diff options
Diffstat (limited to 'model-optimizer/extensions/middle/AddIsCyclicAttribute.py')
-rw-r--r-- | model-optimizer/extensions/middle/AddIsCyclicAttribute.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/model-optimizer/extensions/middle/AddIsCyclicAttribute.py b/model-optimizer/extensions/middle/AddIsCyclicAttribute.py new file mode 100644 index 000000000..c2616ad4d --- /dev/null +++ b/model-optimizer/extensions/middle/AddIsCyclicAttribute.py @@ -0,0 +1,25 @@ +""" + Copyright (c) 2018 Intel Corporation + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. +""" + +import networkx as nx +from mo.middle.replacement import MiddleReplacementPattern + + +class AddIsCyclicAttribute(MiddleReplacementPattern): + @staticmethod + def find_and_replace_pattern(graph: nx.MultiDiGraph): + is_acyclic = nx.is_directed_acyclic_graph(graph) + graph.graph['is_cyclic'] = not is_acyclic
\ No newline at end of file |