Searched refs:allocate_arrays (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 110 void Literal::SetPiece(const Shape& shape, Piece* piece, bool allocate_arrays) { in SetPiece() argument
118 SetPiece(subshape, &child_piece, allocate_arrays); in SetPiece()
123 if (allocate_arrays) { in SetPiece()
145 Literal::Literal(const Shape& shape, bool allocate_arrays) in Literal() argument
153 SetPiece(*shape_, root_piece_, allocate_arrays); in Literal()
|
| D | literal.h | 767 Literal(const Shape& shape, bool allocate_arrays);
792 void SetPiece(const Shape& shape, Piece* piece, bool allocate_arrays);
|