diff options
Diffstat (limited to 'tensorflow/core/framework/allocator.h')
-rw-r--r-- | tensorflow/core/framework/allocator.h | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/framework/allocator.h b/tensorflow/core/framework/allocator.h index 97a3f61693..30c7c19102 100644 --- a/tensorflow/core/framework/allocator.h +++ b/tensorflow/core/framework/allocator.h @@ -187,13 +187,14 @@ class Allocator { // is_simple<T>::value if T[] can be safely constructed and destructed // without running T() and ~T(). We do not use std::is_trivial<T> - // directly because std::complex<float> is not trival but its array - // can be constructed and destructed without running its default ctor - // and dtor. + // directly because std::complex<float> and std::complex<double> are + // not trival, but their arrays can be constructed and destructed + // without running their default ctors and dtors. template <typename T> struct is_simple { static const bool value = std::is_trivial<T>::value || std::is_same<T, complex64>::value || + std::is_same<T, complex128>::value || is_quantized<T>::value; }; |