aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/framework/allocator.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/framework/allocator.h')
-rw-r--r--tensorflow/core/framework/allocator.h7
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;
};