softmax() got an unexpected keyword argument 'axis'

[hananabilabd]

Describe the Issue

I've installed your TensorFlow version 1.1.0 with pip and keras 2.2

 x=Dense(4,activation='softmax',kernel_initializer=glorot_uniform(seed=0))(x)
  File "/usr/local/lib/python2.7/dist-packages/keras/engine/topology.py", line 619, in __call__
    output = self.call(inputs, **kwargs)
  File "/usr/local/lib/python2.7/dist-packages/keras/layers/core.py", line 881, in call
    output = self.activation(output)
  File "/usr/local/lib/python2.7/dist-packages/keras/activations.py", line 29, in softmax
    return K.softmax(x)
  File "/usr/local/lib/python2.7/dist-packages/keras/backend/tensorflow_backend.py", line 2963, in softmax
    return tf.nn.softmax(x, axis=axis)
TypeError: softmax() got an unexpected keyword argument 'axis'

Steps to Reproduce

Hardware/Software Info

Please provide the following information about your Raspberry Pi setup:

• Raspberry Pi model: 3
• Operating System used: Raspberian
• Version of Python used: 2.7
• SD card memory size: 32
• Size of USB/other device used as swap (if building from source):
• TensorFlow git commit hash (if building from source):

Relevant Console Output/Logs