Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h68 const Shape& shape() const { return root_piece().subshape(); } in shape()
526 Piece* piece = &const_cast<Piece&>(root_piece()); in piece()
536 virtual const Piece& root_piece() const = 0;
701 Piece& root_piece() const override { return *root_piece_; }; in root_piece() function
832 const Piece& root_piece() const override { return *root_piece_; }; in root_piece() function
859 const Piece& root_piece() const override { return root_piece_; }; in root_piece() function
928 return root_piece().Get<NativeT>(multi_index); in Get()
941 return root_piece().Set<NativeT>(multi_index, value); in Set()
1067 auto root_data = root_piece().data<NativeT>(); in PopulateSparse()
1072 *this->root_piece().sparse_indices() = std::move(indices); in PopulateSparse()
[all …]
Dliteral.cc1526 return root_piece().ForEachSubpieceWithBool( in operator ==()
1556 return root_piece().ForEachSubpieceWithBool([&](const ShapeIndex& index, in IsAll()
1624 return root_piece().ForEachSubpieceWithBool( in IsAllFloat()
1650 return AllElementsEqualValue<complex64>(root_piece().data<complex64>(), in IsAllComplex()
1653 return AllElementsEqualValue<complex128>(root_piece().data<complex128>(), in IsAllComplex()
1661 return root_piece().ForEachSubpieceWithBool( in IsAllFirst()
2083 root_piece().ForEachSubpiece( in ToProto()
2162 CopyPieceSubtree(*shape_, &literal.root_piece(), root_piece_); in MutableBorrowingLiteral()
2173 CopyPieceSubtree(*shape_, &literal.root_piece(), root_piece_); in operator =()
2187 CopyPieceSubtree(*shape_, &literal.root_piece(), root_piece_); in MutableBorrowingLiteral()
[all …]