[CIBUILD] change the size of optimizers_test

Author: fuhailin

Diff for: .devcontainer/devcontainer.json

@@ -1,5 +1,5 @@
 {
-	"name": "Tensorflow Addons SIG Dev Container",
+	"name": "Tensorflow Deepray SIG Dev Container",
 	// Uncomment this for GPU image
 	// "build": { 
 	// 		"args": { 

Diff for: .github/ISSUE_TEMPLATE/bug-performance-report.md

@@ -7,7 +7,7 @@ about: Use this template for reporting a bug or a performance issue.
 **System information**
 - OS Platform and Distribution (e.g., Linux Ubuntu 16.04):
 - TensorFlow version and how it was installed (source or binary):
-- TensorFlow-Addons version and how it was installed (source or binary):
+- TensorFlow-Deepray version and how it was installed (source or binary):
 - Python version:
 - Is GPU used? (yes/no):
 

Diff for: CONTRIBUTING.md

@@ -26,17 +26,17 @@ review.
 
 ## Requirements for New Contributions to the Repository
 
-**All new components/features to Addons need to first be submitted as a feature 
+**All new components/features to Deepray need to first be submitted as a feature 
 request issue. This will allow the team to check with our counterparts in the TF
 ecosystem and ensure it is not roadmapped internally for Keras or TF core. These 
 feature requests will be labeled with `ecosystem-review` while we determine if it 
-should be included in Addons.**
+should be included in Deepray.**
 
 The tensorflow/deepray repository contains additional functionality
 fitting the following criteria:
 
 * The functionality is not otherwise available in TensorFlow
-* Addons have to be compatible with TensorFlow 2.x.
+* Deepray have to be compatible with TensorFlow 2.x.
 * The addon conforms to the code and documentation standards
 * The addon is impactful to the community (e.g. an implementation used
  in widely cited paper)
@@ -178,7 +178,7 @@ pip install tensorflow==2.12.0
 pip install -e ./
 ```
 
-It's going to install Addons in editable mode without compiling anything.
+It's going to install Deepray in editable mode without compiling anything.
 You can modify source files and changes will be seen at the next Python 
 interpreter startup. This command needs to be executed only once. 
 Now, anywhere on your system, if you do `import deepray`, it's 
@@ -379,7 +379,7 @@ context, see [this pull request](https://github.com/tensorflow/deepray/pull/928)
 You can import some common types 
 from [deepray/utils/types.py](deepray/utils/types.py).
 
-We recommend adding types if you add a new class/function to Addons' public API, 
+We recommend adding types if you add a new class/function to Deepray' public API, 
 but we don't enforce it.
 
 Since adding type hints can be hard, especially for people who are not

Diff for: MIGRATION_TO_CORE.md

@@ -1,7 +1,7 @@
 #### C++
 C++ code should conform to [Google C++ Style Guide](https://google.github.io/styleguide/cppguide.html).
 
-Addons uses [clang-format](https://clang.llvm.org/docs/ClangFormat.html)
+Deepray uses [clang-format](https://clang.llvm.org/docs/ClangFormat.html)
 to check your C/C++ changes. Sometimes you have some manually formatted
 code that you don’t want clang-format to touch.
 You can disable formatting like this:
@@ -57,4 +57,4 @@ yapf --style=./.yapf -ir ./**/*.py
 
 Follow the guidance in the [TensorFlow Style Guide - Conventions](https://www.tensorflow.org/community/contribute/code_style#tensorflow_conventions_and_special_uses).
 
-Please note that Addons follows the conventions of the TensorFlow library, but formats our code using [PEP8](https://www.python.org/dev/peps/pep-0008/) guidelines.
+Please note that Deepray follows the conventions of the TensorFlow library, but formats our code using [PEP8](https://www.python.org/dev/peps/pep-0008/) guidelines.

Diff for: STYLE_GUIDE.md

@@ -1,4 +1,4 @@
-# Addons Build Definitions inherited from TensorFlow Core
+# Deepray Build Definitions inherited from TensorFlow Core
 
 D_GLIBCXX_USE_CXX11_ABI = "%{tf_cx11_abi}"
 CPLUSPLUS_VERSION = "%{tf_cplusplus_ver}"

Diff for: build_deps/tf_dependency/build_defs.bzl.tpl

@@ -1,4 +1,4 @@
-# Addons - Activations
+# Deepray - Activations
 
 ## Contents
 https://www.tensorflow.org/deepray/api_docs/python/dp/activations

Diff for: deepray/activations/README.md

@@ -1,4 +1,4 @@
-# Addons - Callbacks
+# Deepray - Callbacks
 
 ## Contents
 https://www.tensorflow.org/deepray/api_docs/python/dp/callbacks

Diff for: deepray/callbacks/README.md

@@ -1,4 +1,4 @@
-# Addons - Custom Ops
+# Deepray - Custom Ops
 
 ## Contents
 | Sub-Package  | Description                             |

Diff for: deepray/custom_ops/README.md

@@ -1,4 +1,4 @@
-# Addons - Image
+# Deepray - Image
 
 ## Components 
 https://www.tensorflow.org/deepray/api_docs/python/dp/image

Diff for: deepray/image/README.md

@@ -174,7 +174,7 @@ def adjust_hsv_in_yiq(
 
     if not options.is_custom_kernel_disabled():
       warnings.warn(
-          "C++/CUDA kernel of `adjust_hsv_in_yiq` will be removed in Addons `0.13`.",
+          "C++/CUDA kernel of `adjust_hsv_in_yiq` will be removed in Deepray `0.13`.",
           DeprecationWarning,
       )
       try:

Diff for: deepray/image/distort_image_ops.py

@@ -1,4 +1,4 @@
-# Addons - Layers
+# Deepray - Layers
 
 ## Components
 https://www.tensorflow.org/deepray/api_docs/python/dp/layers

Diff for: deepray/layers/README.md

@@ -98,7 +98,7 @@ def __init__(
       **kwargs,
   ):
     warnings.warn(
-        "`MultiHeadAttention` will be deprecated in Addons 0.13. "
+        "`MultiHeadAttention` will be deprecated in Deepray 0.13. "
         "Please use `tf.keras.layers.MultiHeadAttention` instead.",
         DeprecationWarning,
     )

Diff for: deepray/layers/multihead_attention.py

@@ -1,4 +1,4 @@
-# Addons - Losses
+# Deepray - Losses
 
 ## Components
 https://www.tensorflow.org/deepray/api_docs/python/dp/losses

Diff for: deepray/losses/README.md

@@ -1,4 +1,4 @@
-# Addons - Metrics
+# Deepray - Metrics
 
 ## Contents
 https://www.tensorflow.org/deepray/api_docs/python/dp/metrics

Diff for: deepray/metrics/README.md

@@ -92,7 +92,7 @@ def __init__(
     if "y_shape" in kwargs:
       warnings.warn(
           "y_shape has been removed, because it's automatically derived,"
-          "and will be deprecated in Addons 0.18.",
+          "and will be deprecated in Deepray 0.18.",
           DeprecationWarning,
       )
 

Diff for: deepray/metrics/r_square.py

@@ -13,7 +13,7 @@ py_library(
 
 py_test(
     name = "optimizers_test",
-    size = "small",
+    size = "medium",
     srcs = glob(["tests/*"]),
     main = "tests/run_all_test.py",
     deps = [

Diff for: deepray/optimizers/BUILD

@@ -1,4 +1,4 @@
-# Addons - Optimizers
+# Deepray - Optimizers
 
 ## Components
 https://www.tensorflow.org/deepray/api_docs/python/dp/optimizers

Diff for: deepray/optimizers/README.md

@@ -81,7 +81,7 @@ def __init__(
     if "weight_decay_rate" in kwargs:
       warnings.warn(
           "weight_decay_rate has been renamed to weight_decay,"
-          "and will be deprecated in Addons 0.18.",
+          "and will be deprecated in Deepray 0.18.",
           DeprecationWarning,
       )
       weight_decay = kwargs["weight_decay_rate"]

Diff for: deepray/optimizers/lamb.py

@@ -14,7 +14,7 @@
 _FALLBACK_WARNING_TEMPLATE = """{}
 
 The {} C++/CUDA custom op could not be loaded.
-For this reason, Addons will fallback to an implementation written
+For this reason, Deepray will fallback to an implementation written
 in Python with public TensorFlow ops. There worst you might experience with
 this is a moderate slowdown on GPU. There can be multiple
 reason for this loading error, one of them may be an ABI incompatibility between
@@ -23,7 +23,7 @@
 shared object file was displayed above.
 
 If you want this warning to disappear, either make sure the TensorFlow installed
-is compatible with this version of Addons, or tell Deepray to
+is compatible with this version of Deepray, or tell Deepray to
 prefer using Python implementations and not custom C++/CUDA ones. You can do that
 by setting the enviornment variable `DEEPRAY_PY_OPS=1`:
 ```bash

Diff for: deepray/options.py

@@ -9,7 +9,7 @@
 def register_all(keras_objects: bool = True, custom_kernels: bool = True) -> None:
   """Register Deepray' objects in TensorFlow global dictionaries.
 
-    When loading a Keras model that has a TF Addons' function, it is needed
+    When loading a Keras model that has a TF Deepray' function, it is needed
     for this function to be known by the Keras deserialization process.
 
     There are two ways to do this, either do
@@ -53,13 +53,13 @@ def register_all(keras_objects: bool = True, custom_kernels: bool = True) -> Non
             Keras objects
             with `tf.keras.utils.register_keras_serializable(package="Deepray")`
             If set to False, doesn't register any Keras objects
-            of Addons in TensorFlow.
+            of Deepray in TensorFlow.
         custom_kernels: boolean, `True` by default. If `True`, loads all
             custom kernels of Deepray with
             `tf.load_op_library("path/to/so/file.so")`. Loading the SO files
             register them automatically. If `False` doesn't load and register
             the shared objects files. Not that it might be useful to turn it off
-            if your installation of Addons doesn't work well with custom ops.
+            if your installation of Deepray doesn't work well with custom ops.
     Returns:
         None
     """
@@ -82,7 +82,7 @@ def register_custom_kernels() -> None:
   if not all_shared_objects:
     raise FileNotFoundError(
         "No shared objects files were found in the custom ops "
-        "directory in Tensorflow Addons, check your installation again, "
+        "directory in Tensorflow Deepray, check your installation again, "
         "or, if you don't need custom ops, call `dp.register_all(custom_kernels=False)`"
         " instead."
     )

Diff for: deepray/register.py

@@ -1,4 +1,4 @@
-# Addons - RNN
+# Deepray - RNN
 
 ## Components
 https://www.tensorflow.org/deepray/api_docs/python/dp/rnn

Diff for: deepray/rnn/README.md

@@ -1,4 +1,4 @@
-# Addons - Seq2seq
+# Deepray - Seq2seq
 
 ## Contents
 https://www.tensorflow.org/deepray/api_docs/python/dp/seq2seq

Diff for: deepray/seq2seq/README.md

@@ -1,4 +1,4 @@
-# Addons - Text
+# Deepray - Text
 
 ## Components 
 https://www.tensorflow.org/deepray/api_docs/python/dp/text

Diff for: deepray/text/README.md

@@ -1 +1 @@
-# Addons Utils
+# Deepray Utils

Diff for: deepray/utils/README.md

@@ -29,7 +29,7 @@ def _check_tf_version():
   """Warn the user if the version of TensorFlow used is not supported.
 
     This is not a check for custom ops compatibility. This check only ensure that
-    we support this TensorFlow version if the user uses only Addons' Python code.
+    we support this TensorFlow version if the user uses only Deepray' Python code.
     """
 
   if "dev" in tf.__version__:
@@ -50,7 +50,7 @@ def _check_tf_version():
     return
 
   warnings.warn(
-      "Tensorflow Addons supports using Python ops for all Tensorflow versions "
+      "Tensorflow Deepray supports using Python ops for all Tensorflow versions "
       "above or equal to {} and strictly below {} (nightly versions are not "
       "supported). \n "
       "The versions of TensorFlow you are currently using is {} and is not "

Diff for: deepray/utils/ensure_tf_install.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Utilities for testing Addons."""
+"""Utilities for testing Deepray."""
 
 import os
 import random

Diff for: deepray/utils/test_utils.py

@@ -68,7 +68,7 @@ pip install artifacts/deepray-*.whl
 
 #### Standardized API within Subpackages
 User experience and project maintainability are core concepts in
-TF-Addons. In order to achieve these we require that our additions
+Deepray. In order to achieve these we require that our additions
 conform to established API patterns seen in core TensorFlow.
 
 #### GPU/CPU Custom-Ops
@@ -77,7 +77,7 @@ a CUDA 10 installation not be found then the op will automatically fall back to
 a CPU implementation.
 
 #### Proxy Maintainership
-Addons has been designed to compartmentalize subpackages and submodules so 
+Deepray has been designed to compartmentalize subpackages and submodules so 
 that they can be maintained by users who have expertise and a vested interest 
 in that component. 
 
@@ -125,7 +125,7 @@ warning.
 
 
 ## Contributing
-TF-Addons is a community led open source project. As such, the project
+Deepray is a community led open source project. As such, the project
 depends on public contributions, bug-fixes, and documentation. Please see 
 [contribution guidelines](https://github.com/tensorflow/deepray/blob/master/CONTRIBUTING.md) 
 for a guide on how to contribute. This project adheres to [TensorFlow's code of conduct](https://github.com/tensorflow/deepray/blob/master/CODE_OF_CONDUCT.md).

Diff for: docs/overview.md

@@ -5,7 +5,7 @@ Deepray welcomes and highly encourages tutorial contributions.
 
 ## How To Contribute
 
-Addons tutorials are created using [Google Colab](https://colab.research.google.com/) 
+Deepray tutorials are created using [Google Colab](https://colab.research.google.com/) 
 and the jupyter notebooks are saved to this directory in the repository. To do 
 this, follow the below steps:
 
@@ -18,4 +18,4 @@ URL boxes so that they match the name of your new example notebook
 4. Follow the guidelines of the template
 5. "Save a copy in GitHub" and select your new branch. The notebook should be 
 named `subpackage_submodule`
-6. Submit the branch as a PR on the TF-Addons GitHub
+6. Submit the branch as a PR on the Deepray GitHub

Diff for: docs/tutorials/README.md

@@ -112,7 +112,7 @@
         "id": "kTlbneoEUKrD"
       },
       "source": [
-        "### Install Tensorflow 2.0 and Tensorflow-Addons"
+        "### Install Tensorflow 2.0 and Deepray"
       ]
     },
     {

Diff for: docs/tutorials/layers_normalizations.ipynb

@@ -66,7 +66,7 @@
       },
       "source": [
         "# Overview\n",
-        "This notebook will demonstrate how to use the Conditional Graident Optimizer from the Addons package."
+        "This notebook will demonstrate how to use the Conditional Graident Optimizer from the Deepray package."
       ]
     },
     {

Diff for: docs/tutorials/optimizers_conditionalgradient.ipynb

@@ -70,7 +70,7 @@
       "source": [
         "## Overview\n",
         "\n",
-        "This tutorial demonstrates the use of Cyclical Learning Rate from the Addons package."
+        "This tutorial demonstrates the use of Cyclical Learning Rate from the Deepray package."
       ]
     },
     {

Diff for: docs/tutorials/optimizers_cyclicallearningrate.ipynb

@@ -66,7 +66,7 @@
       "source": [
         "## Overview\n",
         "\n",
-        "This notebook will demonstrate how to use the lazy adam optimizer from the Addons package.\n"
+        "This notebook will demonstrate how to use the lazy adam optimizer from the Deepray package.\n"
       ]
     },
     {

Diff for: docs/tutorials/optimizers_lazyadam.ipynb

@@ -108,7 +108,7 @@ def test_no_private_tf_api():
           "A private tensorflow API import was found in {} at line {}.\n"
           "tensorflow.python refers to TensorFlow's internal source "
           "code and private functions/classes.\n"
-          "The use of those is forbidden in Addons for stability reasons."
+          "The use of those is forbidden in Deepray for stability reasons."
           "\nYou should find a public alternative or ask the "
           "TensorFlow team to expose publicly the function/class "
           "that you are using.\n"