Searched refs:SetPiece (Results 1 – 2 of 2) sorted by relevance
110 void Literal::SetPiece(const Shape& shape, Piece* piece, bool allocate_arrays) { in SetPiece() function in xla::Literal118 SetPiece(subshape, &child_piece, allocate_arrays); in SetPiece()153 SetPiece(*shape_, root_piece_, allocate_arrays); in Literal()
792 void SetPiece(const Shape& shape, Piece* piece, bool allocate_arrays);