Home
last modified time | relevance | path

Searched refs:src_buf_ptrs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral.cc2646 MutableBorrowingLiteral::MutableBorrowingLiteral(absl::Span<char*> src_buf_ptrs, in MutableBorrowingLiteral() argument
2651 CHECK_EQ(src_buf_ptrs.size(), 1); in MutableBorrowingLiteral()
2654 root_piece_->set_buffer(const_cast<char*>(src_buf_ptrs[0])); in MutableBorrowingLiteral()
2657 CHECK_EQ(src_buf_ptrs.size(), ShapeUtil::TupleElementCount(*shape_)); in MutableBorrowingLiteral()
2661 for (int i = 0; i < src_buf_ptrs.size(); ++i) { in MutableBorrowingLiteral()
2666 child_piece.set_buffer(src_buf_ptrs[i]); in MutableBorrowingLiteral()
2711 BorrowingLiteral::BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs, in BorrowingLiteral() argument
2716 CHECK_EQ(src_buf_ptrs.size(), ShapeUtil::TupleElementCount(*shape_)); in BorrowingLiteral()
2721 for (int i = 0, end = src_buf_ptrs.size(); i < end; ++i) { in BorrowingLiteral()
2724 root_piece_.child(i).set_buffer(const_cast<char*>(src_buf_ptrs[i])); in BorrowingLiteral()
H A Dliteral_test.cc1701 std::vector<const char*> src_buf_ptrs; in TEST_F() local
1702 src_buf_ptrs.emplace_back( in TEST_F()
1704 src_buf_ptrs.emplace_back(reinterpret_cast<const char*>(hundred.data())); in TEST_F()
1706 src_buf_ptrs, in TEST_F()
H A Dliteral.h1135 MutableBorrowingLiteral(absl::Span<char*> src_buf_ptrs, const Shape& shape);
1175 BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs,