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

Merge pull request #26858 from ROCmSoftwarePlatform:google_upstream_nn_ops_test

Browse Source 
PiperOrigin-RevId: 239862927
This commit is contained in:
TensorFlower Gardener 2019-03-22 15:24:09 -07:00
commit 1d31242d76
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/nn_ops_test.cc
View File

@ -15,9 +15,9 @@ limitations under the License.
#define EIGEN_USE_THREADS #define EIGEN_USE_THREADS
#if GOOGLE_CUDA #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#define EIGEN_USE_GPU #define EIGEN_USE_GPU
#endif // GOOGLE_CUDA #endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#include <functional> #include <functional>
#include <memory> #include <memory>