message: The kernel does not build with ROCm so disable it in ROCm for now
  Doing this to make the ROCm build green.
diff: @@ -13,7 +13,8 @@ See the License for the specific language governing permissions and
  limitations under the License.
  ==============================================================================*/
  -#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  +// The kernel as written below does not build for ROCm.
  +#if GOOGLE_CUDA
  #define EIGEN_USE_GPU
  @@ -470,4 +471,4 @@ REGISTER_UNIQUE_GPU(bool);
  } // end namespace tensorflow
  -#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  +#endif // GOOGLE_CUDA
  