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

Add a missing template explicit instantiation of SetZeroFunctor (#12791)

Browse Source
This commit is contained in:
Changming Sun 2017-09-12 12:36:24 -05:00 committed by gunan
parent 7bb08f5bf2
commit cbc46a8561
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/fill_functor.cc
View File

@ -20,6 +20,7 @@ limitations under the License.
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/framework/tensor_types.h" #include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/framework/types.h" #include "tensorflow/core/framework/types.h"
#include "tensorflow/core/framework/variant_encode_decode.h"
namespace tensorflow { namespace tensorflow {
namespace functor { namespace functor {
@ -50,6 +51,7 @@ DEFINE_SETZERO_CPU(int32);
DEFINE_SETZERO_CPU(int64); DEFINE_SETZERO_CPU(int64);
DEFINE_SETZERO_CPU(complex64); DEFINE_SETZERO_CPU(complex64);
DEFINE_SETZERO_CPU(complex128); DEFINE_SETZERO_CPU(complex128);
DEFINE_SETZERO_CPU(Variant);
#undef DEFINE_SETZERO_CPU #undef DEFINE_SETZERO_CPU
#ifdef TENSORFLOW_USE_SYCL #ifdef TENSORFLOW_USE_SYCL