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

Merge pull request #40585 from yongtang:40577-tensor_scatter_nd_add-gpu

Browse Source 
PiperOrigin-RevId: 320129620
Change-Id: I76c1f633c341ef931be8d3eb2d378ad3e7f12fda
This commit is contained in:
TensorFlower Gardener 2020-07-07 22:57:21 -07:00
commit 014cdf65ab
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/inplace_ops_functor_gpu.cu.cc
View File

@ -197,6 +197,8 @@ Status DoCopy(const Device& d, const Tensor& x, Tensor* y) {
CASE(float)
CASE(double)
CASE(Eigen::half)
CASE(complex64)
CASE(complex128)
CASE(int64)
#undef CASE
default: