Updated the naming from 'tf.argmax' to 'tf.math.argmax'

Author: tilakrayal

site/en/guide/migrate/metrics_optimizers.ipynb

@@ -144,7 +144,7 @@
  "\n",
  "def _model_fn(features, labels, mode):\n",
  " logits = tf1.layers.Dense(2)(features)\n",
- " predictions = tf.argmax(input=logits, axis=1)\n",
+ " predictions = tf.math.argmax(input=logits, axis=1)\n",
  " loss = tf1.nn.sparse_softmax_cross_entropy_with_logits(labels=labels, logits=logits)\n",
  " optimizer = tf1.train.AdagradOptimizer(0.05)\n",
  " train_op = optimizer.minimize(loss, global_step=tf1.train.get_global_step())\n",
@@ -223,7 +223,7 @@
  "\n",
  "inputs = tf.keras.Input((2,))\n",
  "logits = tf.keras.layers.Dense(2)(inputs)\n",
- "predictions = tf.argmax(input=logits, axis=1)\n",
+ "predictions = tf.math.argmax(input=logits, axis=1)\n",
  "model = tf.keras.models.Model(inputs, predictions)\n",
  "optimizer = tf.keras.optimizers.Adagrad(learning_rate=0.05)\n",
  "\n",