def _get_unimplemented_ops(self, pb_path):
"""
Returns a list of unimplemented ops' names.
Arguments:
pb_path: Protobuf file path.
Returns:
List of unimplemented ops' names.
"""
# get required op
with open(pb_path) as f:
lines = f.readlines()
lines = [l.strip() for l in lines]
required_ops = set()
for line in lines:
if line[:3] == 'op:':
op_name = line.split(' ')[1][1:-1]
required_ops.add(op_name)
# get supported ops
ob = OpsBridge()
supported_ops = set([
name for name in dir(ob)
if name[:1] != "_" and name not in ob.__dict__
])
# get unimplemented ops
unimplemented_ops = required_ops - supported_ops
return sorted(list(unimplemented_ops))
def reset(self):
"""
Resets importer states.
"""
self.name_op_map = dict()
self.ops_bridge = OpsBridge()
self.init_ops = []
self.graph_def = None
def reset(self):
"""
Resets importer states.
"""
# TF's graph and graph_def
self._graph = None
self._graph_def = None
# name to op dict and obs bridge converter
self._name_op_map = dict()
self._ops_bridge = OpsBridge()
# checkpoint path for weight import
self._checkpoint_path = None