diff options
Diffstat (limited to 'tensorflow/compiler/xla/literal_util.h')
-rw-r--r-- | tensorflow/compiler/xla/literal_util.h | 25 |
1 files changed, 1 insertions, 24 deletions
diff --git a/tensorflow/compiler/xla/literal_util.h b/tensorflow/compiler/xla/literal_util.h index f37e529caf..a1e288829f 100644 --- a/tensorflow/compiler/xla/literal_util.h +++ b/tensorflow/compiler/xla/literal_util.h @@ -163,11 +163,6 @@ class Literal { const std::vector<complex64>& c64s() const { return c64s_; } std::vector<complex64>* mutable_c64s() { return &c64s_; } - int bf16s_size() const { return bf16s().size(); } - bfloat16 bf16s(int i) const { return bf16s_[i]; } - const std::vector<bfloat16>& bf16s() const { return bf16s_; } - std::vector<bfloat16>* mutable_bf16s() { return &bf16s_; } - int tuple_literals_size() const { return tuple_literals().size(); } const Literal& tuple_literals(int i) const { return tuple_literals_[i]; } Literal* add_tuple_literals() { @@ -455,7 +450,7 @@ class Literal { tensorflow::Status ValidateLiteral() const; // Returns a string representation of the literal value. - string ToString(bool print_layout = false) const; + string ToString() const; // Invokes the "per cell" callback for each element in the provided // literal with the element's indices and a string representation of @@ -627,7 +622,6 @@ class Literal { std::vector<uint16> u16s_; std::vector<uint32> u32s_; std::vector<uint64> u64s_; - std::vector<bfloat16> bf16s_; std::vector<half> f16s_; std::vector<float> f32s_; std::vector<double> f64s_; @@ -681,9 +675,6 @@ template <> tensorflow::gtl::ArraySlice<half> Literal::GetArraySlice<half>() const; template <> -tensorflow::gtl::ArraySlice<bfloat16> Literal::GetArraySlice<bfloat16>() const; - -template <> tensorflow::gtl::ArraySlice<complex64> Literal::GetArraySlice<complex64>() const; @@ -724,9 +715,6 @@ template <> tensorflow::gtl::MutableArraySlice<half> Literal::GetMutableArraySlice(); template <> -tensorflow::gtl::MutableArraySlice<bfloat16> Literal::GetMutableArraySlice(); - -template <> tensorflow::gtl::MutableArraySlice<complex64> Literal::GetMutableArraySlice(); template <> @@ -760,9 +748,6 @@ template <> void Literal::Resize<half>(int64 num_elements, half value); template <> -void Literal::Resize<bfloat16>(int64 num_elements, bfloat16 value); - -template <> void Literal::Resize<complex64>(int64 num_elements, complex64 value); template <typename NativeT> @@ -1005,14 +990,6 @@ inline half Literal::Get<half>( return GetArraySlice<half>()[linear_index]; } -template <> -inline bfloat16 Literal::Get<bfloat16>( - tensorflow::gtl::ArraySlice<int64> multi_index) const { - CHECK(shape().element_type() == BF16); - int64 linear_index = LinearIndex(multi_index); - return GetArraySlice<bfloat16>()[linear_index]; -} - template <typename NativeT> void Literal::Set(tensorflow::gtl::ArraySlice<int64> multi_index, NativeT value) { |