Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/framework/op_def_registry.py: 36%
25 statements
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
1# Copyright 2015 The TensorFlow Authors. All Rights Reserved.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14# ==============================================================================
16"""Global registry for OpDefs."""
18import threading
20from tensorflow.core.framework import op_def_pb2
21# pylint: disable=invalid-import-order,g-bad-import-order, wildcard-import, unused-import
22from tensorflow.python import pywrap_tensorflow
23from tensorflow.python.framework import _op_def_registry
25# The cache amortizes ProtoBuf serialization/deserialization overhead
26# on the language boundary. If an OpDef has been looked up, its Python
27# representation is cached.
28_cache = {}
29_cache_lock = threading.Lock()
32def get(name):
33 """Returns an OpDef for a given `name` or None if the lookup fails."""
34 try:
35 return _cache[name]
36 except KeyError:
37 pass
39 with _cache_lock:
40 try:
41 # Return if another thread has already populated the cache.
42 return _cache[name]
43 except KeyError:
44 pass
46 serialized_op_def = _op_def_registry.get(name)
47 if serialized_op_def is None:
48 return None
50 op_def = op_def_pb2.OpDef()
51 op_def.ParseFromString(serialized_op_def)
52 _cache[name] = op_def
53 return op_def
56# TODO(b/141354889): Remove once there are no callers.
57def sync():
58 """No-op. Used to synchronize the contents of the Python registry with C++."""