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

cbc46a85 · Changming Sun · gunan · 7bb08f5b · cbc46a85
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

tensorflow/core/kernels/fill_functor.cc tensorflow/core/kernels/fill_functor.cc +2 -0

未找到文件。
--- a/tensorflow/core/kernels/fill_functor.cc
+++ b/tensorflow/core/kernels/fill_functor.cc
@@ -20,6 +20,7 @@ limitations under the License.
 #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
 #include "tensorflow/core/framework/tensor_types.h"
 #include "tensorflow/core/framework/types.h"
+#include "tensorflow/core/framework/variant_encode_decode.h"

 namespace tensorflow {
 namespace functor {
@@ -50,6 +51,7 @@ DEFINE_SETZERO_CPU(int32);
 DEFINE_SETZERO_CPU(int64);
 DEFINE_SETZERO_CPU(complex64);
 DEFINE_SETZERO_CPU(complex128);
+DEFINE_SETZERO_CPU(Variant);
 #undef DEFINE_SETZERO_CPU

 #ifdef TENSORFLOW_USE_SYCL