experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix ROCM build

Browse Source 
PiperOrigin-RevId: 276330686
Change-Id: I2b9ff29ce7e35280d92631ac9f7e730b0dce9c6d
This commit is contained in:
Eugene Zhulenev 2019-10-23 12:43:49 -07:00 committed by TensorFlower Gardener
parent 52354aaa7f
commit b07081a4df
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/conv_grad_input_ops.cc
View File

@ -217,7 +217,7 @@ struct LaunchConv2DBackpropInputOp<CPUDevice, T> {
} }
}; };
#ifdef GOOGLE_CUDA #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
// Computes backprop input using Eigen::SpatialConvolutionBackwardInput on GPU // Computes backprop input using Eigen::SpatialConvolutionBackwardInput on GPU
// for int32 inputs. // for int32 inputs.
template <> template <>
@ -234,7 +234,7 @@ struct LaunchConv2DBackpropInputOp<GPUDevice, int32> {
explicit_paddings, in_backprop, data_format); explicit_paddings, in_backprop, data_format);
} }
}; };
#endif // GOOGLE_CUDA #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#ifdef TENSORFLOW_USE_LIBXSMM_CONVOLUTIONS #ifdef TENSORFLOW_USE_LIBXSMM_CONVOLUTIONS
template <typename Device, class T> template <typename Device, class T>