Updated ordering for kwargs

1 files changed, 3 insertions, 3 deletions

diff --git a/tensorflow/python/layers/core.py b/tensorflow/python/layers/core.py
index 5919fa543e..e06e9aba4a 100644
--- a/tensorflow/python/layers/core.py
+++ b/tensorflow/python/layers/core.py
@@ -292,17 +292,17 @@ class Flatten(keras_layers.Flatten, base.Layer):
 
 
 @tf_export('layers.flatten')
-def flatten(inputs, data_format='channels_last', name=None):
+def flatten(inputs, name=None, data_format='channels_last'):
   """Flattens an input tensor while preserving the batch axis (axis 0).
 
   Arguments:
     inputs: Tensor input.
+    name: The name of the layer (string).
     data_format: A string, one of `channels_last` (default) or `channels_first`.
       The ordering of the dimensions in the inputs.
       `channels_last` corresponds to inputs with shape
       `(batch, height, width, channels)` while `channels_first` corresponds to
       inputs with shape `(batch, channels, height, width)`.
-    name: The name of the layer (string).
 
   Returns:
     Reshaped tensor.
@@ -319,7 +319,7 @@ def flatten(inputs, data_format='channels_last', name=None):
     # now `y` has shape `(None, None)`
   ```
   """
-  layer = Flatten(data_format=data_format, name=name)
+  layer = Flatten(name=name, data_format=data_format)
   return layer.apply(inputs)