Source code for mxnet.attribute

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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.

# coding: utf-8
"""Attribute scoping support for symbolic API."""
import threading
import warnings
from collections import defaultdict

from .base import string_types, classproperty, with_metaclass, _MXClassPropertyMetaClass

[docs]class AttrScope(with_metaclass(_MXClassPropertyMetaClass, object)): """Attribute manager for scoping. User can also inherit this object to change naming behavior. Parameters ---------- kwargs The attributes to set for all symbol creations in the scope. """ _current = threading.local() _subgraph_names = defaultdict(int) def __init__(self, **kwargs): self._old_scope = None for value in kwargs.values(): if not isinstance(value, string_types): raise ValueError("Attributes need to be string") self._attr = kwargs
[docs] def get(self, attr): """ Get the attribute dict given the attribute set by the symbol. Parameters ---------- attr : dict of string to string The attribute passed in by user during symbol creation. Returns ------- attr : dict of string to string Updated attributes to add other scope related attributes. """ if self._attr: ret = self._attr.copy() if attr: ret.update(attr) return ret else: return attr if attr else {}
def __enter__(self): # pylint: disable=protected-access if not hasattr(AttrScope._current, "value"): AttrScope._current.value = AttrScope() self._old_scope = AttrScope._current.value attr = AttrScope._current.value._attr.copy() attr.update(self._attr) self._attr = attr AttrScope._current.value = self return self def __exit__(self, ptype, value, trace): assert self._old_scope AttrScope._current.value = self._old_scope #pylint: disable=no-self-argument @classproperty def current(cls): warnings.warn("AttrScope.current has been deprecated. " "It is advised to use the `with` statement with AttrScope.", DeprecationWarning) if not hasattr(AttrScope._current, "value"): cls._current.value = AttrScope() return cls._current.value @current.setter def current(cls, val): warnings.warn("AttrScope.current has been deprecated. " "It is advised to use the `with` statement with AttrScope.", DeprecationWarning) cls._current.value = val
#pylint: enable=no-self-argument AttrScope._current.value = AttrScope()