message: Instantiate conv templates for ROCM
diff: @@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
  limitations under the License.
  ==============================================================================*/
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #define EIGEN_USE_GPU
  @@ -42,4 +42,4 @@ template struct SpatialConvolutionBackwardInputWithExplicitPaddingFunc<
  } // namespace functor
  } // namespace tensorflow
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  