Update simple_layer input to use tf.constant

[Kayyuri]

Modified the input of simple_layer to use tf.constant.
Fixes: #115674

[github-actions]

Preview

Preview and run these notebook edits with Google Colab:

• site/en/guide/intro_to_modules.ipynb

Rendered notebook diffs available on ReviewNB.com.

Format and style

Use the TensorFlow docs notebook tools to format for consistent source diffs and lint for style:

$ python3 -m pip install -U --user git+https://github.com/tensorflow/docs

$ python3 -m tensorflow_docs.tools.nbfmt notebook.ipynb

$ python3 -m tensorflow_docs.tools.nblint --arg=repo:tensorflow/docs notebook.ipynb

If commits are added to the pull request, synchronize your local branch: git pull origin patch-2