diff options
Diffstat (limited to 'tensorflow/cc/framework/ops.h')
-rw-r--r-- | tensorflow/cc/framework/ops.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/cc/framework/ops.h b/tensorflow/cc/framework/ops.h index a085e1d6e2..0717e7dd4b 100644 --- a/tensorflow/cc/framework/ops.h +++ b/tensorflow/cc/framework/ops.h @@ -150,7 +150,7 @@ class Input { Initializer(const std::initializer_list<T>& v, const TensorShape& shape) { typedef typename RealType<T>::type RealT; Tensor t(DataTypeToEnum<RealT>::v(), shape); - if (t.NumElements() != v.size()) { + if (t.NumElements() != static_cast<int64>(v.size())) { status = errors::InvalidArgument( "Cannot construct a tensor with ", t.NumElements(), " from an initializer list with ", v.size(), " elements"); |