diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index d2d56a880a..748f12ccdb 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -42,20 +42,24 @@ jobs:
         # https://github.com/bazelbuild/bazel/issues/14232#issuecomment-1011247429
         os: ['macos-10.15', 'windows-2019', 'ubuntu-18.04']
         py-version: ['3.7', '3.8', '3.9', '3.10']
-        tf-version: ['2.7.2', '2.8.1', '2.9.0']
+        tf-version: ['2.7.3', '2.8.2', '2.9.1']
         cpu: ['x86']
         exclude:
           - py-version: '3.10'
-            tf-version: '2.7.2'
+            tf-version: '2.7.3'
         include:
           - os: 'macos-11'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.8'
           - os: 'macos-11'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.9'
+          - os: 'macos-11'
+            cpu: 'arm64'
+            tf-version: '2.9.1'
+            py-version: '3.10'
       fail-fast: false
     steps:
       - uses: actions/github-script@0.3.0
@@ -97,17 +101,21 @@ jobs:
       matrix:
         os: ['macOS', 'Windows', 'Linux']
         py-version: ['3.7', '3.8', '3.9', '3.10']
-        tf-version: ['2.9.0']
+        tf-version: ['2.9.1']
         cpu: ['x86']
         include:
           - os: 'macOS'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.8'
           - os: 'macOS'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.9'
+          - os: 'macOS'
+            cpu: 'arm64'
+            tf-version: '2.9.1'
+            py-version: '3.10'
       fail-fast: false
     if: (github.event_name == 'push' && github.ref == 'refs/heads/master') || github.event_name == 'release'
     steps:
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
index d31cc15c7b..8762f0f3f7 100644
--- a/CONTRIBUTING.md
+++ b/CONTRIBUTING.md
@@ -174,7 +174,7 @@ Just run from the root:
 
 ```bash
 pip install tensorflow==2.8
-# you can use "pip install tensorflow-cpu==2.9.0" too if you're not testing on gpu.
+# you can use "pip install tensorflow-cpu==2.9.1" too if you're not testing on gpu.
 pip install -e ./
 ```
 
diff --git a/README.md b/README.md
index a9af00d79e..08eda66cfc 100644
--- a/README.md
+++ b/README.md
@@ -81,6 +81,7 @@ what it was tested against.
 | TensorFlow Addons | TensorFlow | Python  |
 |:----------------------- |:---|:---------- |
 | tfa-nightly | 2.7, 2.8, 2.9 | 3.7, 3.8, 3.9, 3.10 | 
+| tensorflow-addons-0.17.0 | 2.7, 2.8, 2.9 |3.7, 3.8, 3.9, 3.10 |
 | tensorflow-addons-0.16.1 | 2.6, 2.7, 2.8 |3.7, 3.8, 3.9, 3.10 |
 | tensorflow-addons-0.15.0 | 2.5, 2.6, 2.7 |3.7, 3.8, 3.9 |
 | tensorflow-addons-0.14.0 | 2.4, 2.5, 2.6 |3.6, 3.7, 3.8, 3.9 |
@@ -112,6 +113,7 @@ is compiled differently. A typical example of this would be `conda`-installed Te
 | TensorFlow Addons | TensorFlow | Compiler  | cuDNN | CUDA | 
 |:----------------------- |:---- |:---------|:---------|:---------|
 | tfa-nightly | 2.9 | GCC 7.3.1 | 8.1 | 11.2 |
+| tensorflow-addons-0.17.0 | 2.9  | GCC 7.3.1 | 8.1 | 11.2 |
 | tensorflow-addons-0.16.1 | 2.8  | GCC 7.3.1 | 8.1 | 11.2 |
 | tensorflow-addons-0.15.0 | 2.7  | GCC 7.3.1 | 8.1 | 11.2 |
 | tensorflow-addons-0.14.0 | 2.6  | GCC 7.3.1 | 8.1 | 11.2 |
diff --git a/tensorflow_addons/image/color_ops.py b/tensorflow_addons/image/color_ops.py
index e8ec8c7ecf..4f4852175e 100644
--- a/tensorflow_addons/image/color_ops.py
+++ b/tensorflow_addons/image/color_ops.py
@@ -79,8 +79,8 @@ def equalize(
     with tf.name_scope(name or "equalize"):
         image_dims = tf.rank(image)
         image = to_4D_image(image)
-        fn = partial(_equalize_image)
-        image = tf.map_fn(lambda x: fn(x, bins), image)
+        fn = partial(_equalize_image, bins=bins)
+        image = tf.map_fn(fn, image)
         return from_4D_image(image, image_dims)
 
 
diff --git a/tensorflow_addons/version.py b/tensorflow_addons/version.py
index 102024e5cd..49858494c2 100644
--- a/tensorflow_addons/version.py
+++ b/tensorflow_addons/version.py
@@ -21,14 +21,14 @@
 # We follow Semantic Versioning (https://semver.org/)
 _MAJOR_VERSION = "0"
 _MINOR_VERSION = "17"
-_PATCH_VERSION = "0"
+_PATCH_VERSION = "1"
 
 # When building releases, we can update this value on the release branch to
 # reflect the current release candidate ('rc0', 'rc1') or, finally, the official
 # stable release (indicated by `_VERSION_SUFFIX = ''`). Outside the context of a
 # release branch, the current version is by default assumed to be a
 # 'development' version, labeled 'dev'.
-_VERSION_SUFFIX = "dev"
+_VERSION_SUFFIX = ""
 
 # Example, '0.1.0-dev'
 __version__ = ".".join([_MAJOR_VERSION, _MINOR_VERSION, _PATCH_VERSION])
diff --git a/tools/build_dev_container.sh b/tools/build_dev_container.sh
index cecf807f26..5b2f111147 100755
--- a/tools/build_dev_container.sh
+++ b/tools/build_dev_container.sh
@@ -4,7 +4,7 @@ set -x -e
 
 docker build \
     -f tools/docker/dev_container.Dockerfile \
-    --build-arg TF_VERSION=2.9.0 \
+    --build-arg TF_VERSION=2.9.1 \
     --build-arg TF_PACKAGE=tensorflow \
     --build-arg PY_VERSION=$PY_VERSION \
     --no-cache \
diff --git a/tools/docker/build_wheel.Dockerfile b/tools/docker/build_wheel.Dockerfile
index 73d228d321..1c2d28818d 100644
--- a/tools/docker/build_wheel.Dockerfile
+++ b/tools/docker/build_wheel.Dockerfile
@@ -1,6 +1,6 @@
 #syntax=docker/dockerfile:1.1.5-experimental
 ARG PY_VERSION
-FROM tensorflow/build:latest-python$PY_VERSION as base_install
+FROM tensorflow/build:2.9-python$PY_VERSION as base_install
 
 ENV TF_NEED_CUDA="1"
 ARG PY_VERSION
diff --git a/tools/docker/cpu_tests.Dockerfile b/tools/docker/cpu_tests.Dockerfile
index 5d8b6a173b..e3b7ec662b 100644
--- a/tools/docker/cpu_tests.Dockerfile
+++ b/tools/docker/cpu_tests.Dockerfile
@@ -1,7 +1,7 @@
 #syntax=docker/dockerfile:1.1.5-experimental
 FROM python:3.7 as build_wheel
 
-ARG TF_VERSION=2.9.0
+ARG TF_VERSION=2.9.1
 RUN pip install --default-timeout=1000 tensorflow-cpu==$TF_VERSION
 
 RUN apt-get update && apt-get install -y sudo rsync
diff --git a/tools/install_deps/tensorflow-cpu.txt b/tools/install_deps/tensorflow-cpu.txt
index 694692e49a..103fd44862 100644
--- a/tools/install_deps/tensorflow-cpu.txt
+++ b/tools/install_deps/tensorflow-cpu.txt
@@ -1 +1 @@
-tensorflow-cpu~=2.9.0
+tensorflow-cpu~=2.9.1
diff --git a/tools/install_deps/tensorflow.txt b/tools/install_deps/tensorflow.txt
index 9473276a8b..eec3f02c37 100644
--- a/tools/install_deps/tensorflow.txt
+++ b/tools/install_deps/tensorflow.txt
@@ -1 +1 @@
-tensorflow~=2.9.0
\ No newline at end of file
+tensorflow~=2.9.1
\ No newline at end of file
diff --git a/tools/run_gpu_tests.sh b/tools/run_gpu_tests.sh
index caffd747b2..59624750f7 100644
--- a/tools/run_gpu_tests.sh
+++ b/tools/run_gpu_tests.sh
@@ -6,7 +6,7 @@ export DOCKER_BUILDKIT=1
 docker build \
        -f tools/docker/build_wheel.Dockerfile \
        --target tfa_gpu_tests \
-       --build-arg TF_VERSION=2.9.0 \
+       --build-arg TF_VERSION=2.9.1 \
        --build-arg PY_VERSION=3.7 \
        -t tfa_gpu_tests ./
 docker run --rm -t --gpus=all tfa_gpu_tests