Skip to main content

Browse the archive

Revision f04806e2ddcd9af66be2bec85a9ec9311f3a31a5 authored by Jean Kossaifi on 23 December 2018, 23:32:26 UTC, committed by Jean Kossaifi on 23 December 2018, 23:32:26 UTC 
MXNET: fix reshape with empty shape
1 parent 05b6b2f
Raw File
__init__.py 
import sys
from .. import _BACKEND

sys.stderr.write('Using {} backend.\n'.format(_BACKEND))

if _BACKEND == 'mxnet':
    from .mxnet_backend import *
elif _BACKEND == 'numpy':
    from .numpy_backend import *
elif _BACKEND == 'pytorch':
    from .pytorch_backend import *
elif _BACKEND == 'tensorflow':
    from .tensorflow_backend import *
elif _BACKEND == 'cupy':
    from .cupy_backend import *
else:
    import warnings
    warnings.warn('_BACKEND should be either "mxnet", "pytorch", '
                  '"tensorflow", "cupy" or "numpy". '
                  '{} was given.'.format(_BACKEND))
    warnings.warn('Using MXNet backend.')
back to top