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

Fix macro typo that prevented SYCL registrations for ScatterNdMin

Browse Source 
and ScatterNdMax.

PiperOrigin-RevId: 323636493
Change-Id: I42237f4b6dadbdc7d6187283b0282eb6dca5e5e5
This commit is contained in:
Victor de Souza 2020-07-28 13:05:06 -07:00 committed by TensorFlower Gardener
parent 33e709ba69
commit 6c9fad23bf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/scatter_nd_op.cc
View File

@ -513,7 +513,7 @@ TF_CALL_COMPLEX_TYPES(REGISTER_SCATTER_ND_ALL_GPU);
#define REGISTER_SCATTER_ND_UPDATE_SYCL(type) \
REGISTER_SCATTER_ND_UPDATE(type, SYCL);
#define REGISTER_SCATTER_ND_MIN_MAX_GPU(type) \
#define REGISTER_SCATTER_ND_MIN_MAX_SYCL(type) \
REGISTER_SCATTER_ND_MIN_MAX(type, SYCL);
TF_CALL_int32(REGISTER_SCATTER_ND_ADD_SUB_SYCL);