Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc150 child_piece.set_subshape(&subshape); in SetPiece()
179 root_piece_->set_subshape(shape_.get()); in Literal()
629 src_literal.root_piece_->set_subshape(src_literal.shape_.get()); in MoveFrom()
2282 child_piece.set_subshape(&subshape); in CopyPieceSubtree()
2307 root_piece_->set_subshape(shape_.get()); in MutableBorrowingLiteral()
2318 root_piece_->set_subshape(shape_.get()); in operator =()
2331 root_piece_->set_subshape(shape_.get()); in MutableBorrowingLiteral()
2343 root_piece_->set_subshape(shape_.get()); in MutableBorrowingLiteral()
2357 root_piece_->set_subshape(shape_.get()); in MutableBorrowingLiteral()
2368 root_piece_->set_subshape(shape_.get()); in MutableBorrowingLiteral()
[all …]
Dliteral.h400 void set_subshape(const Shape* subshape) { subshape_ = subshape; } in set_subshape() function