Home
last modified time | relevance | path

Searched refs:root_piece_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc149 root_piece_ = new Piece(); in Literal()
150 root_piece_->set_subshape(shape_.get()); in Literal()
151 CHECK(&root_piece_->subshape() == shape_.get()); in Literal()
153 SetPiece(*shape_, root_piece_, allocate_arrays); in Literal()
157 if (root_piece_ != nullptr) { in ~Literal()
159 delete root_piece_; in ~Literal()
164 root_piece_->ForEachMutableSubpiece( in DeallocateBuffers()
178 DCHECK(&other.root_piece_->subshape() == other.shape_.get()); in operator =()
181 swap(root_piece_, other.root_piece_); in operator =()
182 DCHECK(&root_piece_->subshape() == shape_.get()); in operator =()
[all …]
Dliteral.h701 Piece& root_piece() const override { return *root_piece_; }; in root_piece()
737 Piece* root_piece_ = nullptr; variable
832 const Piece& root_piece() const override { return *root_piece_; }; in root_piece()
834 const Piece* root_piece_; // Not owned. variable
859 const Piece& root_piece() const override { return root_piece_; }; in root_piece()
860 Piece root_piece_; variable