update _scipy_fftpack.py

Author: vtavana

mkl_fft/_scipy_fftpack.py

@@ -24,47 +24,47 @@
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-from . import _float_utils
 from . import _pydfti as mkl_fft  # pylint: disable=no-name-in-module
+from ._float_utils import __upcast_float16_array
 
 __all__ = ["fft", "ifft", "fftn", "ifftn", "fft2", "ifft2", "rfft", "irfft"]
 
 
 def fft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.fft(x, n=n, axis=axis, overwrite_x=overwrite_x)
 
 
 def ifft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.ifft(x, n=n, axis=axis, overwrite_x=overwrite_x)
 
 
 def fftn(a, shape=None, axes=None, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.fftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.fftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def ifftn(a, shape=None, axes=None, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.ifftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.ifftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def fft2(a, shape=None, axes=(-2, -1), overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.fftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.fftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def ifft2(a, shape=None, axes=(-2, -1), overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
-    return mkl_fft.ifftn(x, shape=shape, axes=axes, overwrite_x=overwrite_x)
+    x = __upcast_float16_array(a)
+    return mkl_fft.ifftn(x, s=shape, axes=axes, overwrite_x=overwrite_x)
 
 
 def rfft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.rfftpack(x, n=n, axis=axis, overwrite_x=overwrite_x)
 
 
 def irfft(a, n=None, axis=-1, overwrite_x=False):
-    x = _float_utils.__upcast_float16_array(a)
+    x = __upcast_float16_array(a)
     return mkl_fft.irfftpack(x, n=n, axis=axis, overwrite_x=overwrite_x)