1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
# Copyright 2017 The TensorFlow Authors. All Rights Reserved.
#
# 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.
# ==============================================================================
"""Symbol naming utilities."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from tensorflow.python.autograph.pyct import qual_names
class Namer(object):
"""Implementation of the namer interfaces required by various converters.
This implementation performs additional tasks like keeping track of the
function calls that have been encountered and replaced with calls to their
corresponding compiled counterparts.
Interfaces currently implemented:
* call_trees.FunctionNamer
* control_flow.SymbolNamer
* side_effect_guards.SymbolNamer
"""
def __init__(self, global_namespace, recursive, name_map, partial_types):
self.global_namespace = global_namespace
self.recursive = recursive
self.partial_types = partial_types
self.renamed_calls = {}
if name_map is not None:
self.renamed_calls.update(name_map)
self.generated_names = set()
def compiled_class_name(self, original_fqn, live_entity=None):
"""See call_trees.FunctionNamer.compiled_class_name."""
if live_entity is not None and live_entity in self.renamed_calls:
return self.renamed_calls[live_entity]
if isinstance(original_fqn, tuple):
original_name = '__'.join(original_fqn)
else:
original_name = original_fqn
new_name_root = 'Tf%s' % original_name
new_name = new_name_root
n = 0
while new_name in self.global_namespace:
n += 1
new_name = '%s_%d' % (new_name_root, n)
self.generated_names.add(new_name)
if live_entity is not None:
self.renamed_calls[live_entity] = new_name
return new_name
def compiled_function_name(self,
original_fqn,
live_entity=None,
owner_type=None):
"""See call_trees.FunctionNamer.compiled_function_name."""
if not self.recursive:
return None, False
if owner_type is not None and owner_type not in self.partial_types:
# Members are not renamed when part of an entire converted class.
return None, False
if isinstance(original_fqn, tuple):
original_name = '__'.join(original_fqn)
else:
original_name = original_fqn
if live_entity is not None and live_entity in self.renamed_calls:
return self.renamed_calls[live_entity], True
new_name_root = 'tf__%s' % original_name
new_name = new_name_root
n = 0
while new_name in self.global_namespace:
n += 1
new_name = '%s_%d' % (new_name_root, n)
if live_entity is not None:
self.renamed_calls[live_entity] = new_name
self.generated_names.add(new_name)
return new_name, True
def new_symbol(self, name_root, reserved_locals):
"""See control_flow.SymbolNamer.new_symbol."""
# reserved_locals may contain QNs.
all_reserved_locals = set()
for s in reserved_locals:
if isinstance(s, qual_names.QN):
all_reserved_locals.update(s.qn)
elif isinstance(s, str):
all_reserved_locals.add(s)
else:
raise ValueError('Unexpected symbol type "%s"' % type(s))
pieces = name_root.split('_')
if pieces[-1].isdigit():
name_root = '_'.join(pieces[:-1])
n = int(pieces[-1])
else:
n = 0
new_name = name_root
while (new_name in self.global_namespace or
new_name in all_reserved_locals or new_name in self.generated_names):
n += 1
new_name = '%s_%d' % (name_root, n)
self.generated_names.add(new_name)
return new_name
|
