message: Added ROCm support for nn_ops_test
diff: @@ -15,9 +15,9 @@ limitations under the License.
  #define EIGEN_USE_THREADS
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #define EIGEN_USE_GPU
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #include <functional>
  #include <memory>
  