Skip to content

Commit

Permalink
Remove MKLDNN array in python.
Browse files Browse the repository at this point in the history
  • Loading branch information
zheng-da committed Jan 13, 2018
1 parent 80eb4e3 commit c7a696c
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 91 deletions.
87 changes: 0 additions & 87 deletions python/mxnet/ndarray/mkldnn.py

This file was deleted.

1 change: 0 additions & 1 deletion python/mxnet/ndarray/ndarray.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,6 @@
_STORAGE_TYPE_DEFAULT = 0
_STORAGE_TYPE_ROW_SPARSE = 1
_STORAGE_TYPE_CSR = 2
_STORAGE_TYPE_MKLDNN = 3

# pylint: disable= no-member
_DTYPE_NP_TO_MX = {
Expand Down
3 changes: 0 additions & 3 deletions python/mxnet/ndarray/sparse.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,6 @@
from ._internal import _set_ndarray_class
from .ndarray import NDArray, _storage_type, _DTYPE_NP_TO_MX, _DTYPE_MX_TO_NP
from .ndarray import _STORAGE_TYPE_STR_TO_ID, _STORAGE_TYPE_ROW_SPARSE, _STORAGE_TYPE_CSR
from .ndarray import _STORAGE_TYPE_MKLDNN
from .ndarray import _STORAGE_TYPE_UNDEFINED, _STORAGE_TYPE_DEFAULT
from .ndarray import zeros as _zeros_ndarray
from .ndarray import array as _array
Expand Down Expand Up @@ -1139,8 +1138,6 @@ def _ndarray_cls(handle, writable=True, stype=_STORAGE_TYPE_UNDEFINED):
stype = _storage_type(handle)
if stype == _STORAGE_TYPE_DEFAULT:
return NDArray(handle, writable=writable)
elif stype == _STORAGE_TYPE_MKLDNN:
return NDArray(handle, writable=writable)
elif stype == _STORAGE_TYPE_CSR:
return CSRNDArray(handle, writable=writable)
elif stype == _STORAGE_TYPE_ROW_SPARSE:
Expand Down

0 comments on commit c7a696c

Please sign in to comment.