Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc2360 MutableBorrowingLiteral::MutableBorrowingLiteral(absl::Span<char*> src_buf_ptrs, in MutableBorrowingLiteral() argument
2365 CHECK_EQ(src_buf_ptrs.size(), 1); in MutableBorrowingLiteral()
2367 root_piece_->set_buffer(const_cast<char*>(src_buf_ptrs[0])); in MutableBorrowingLiteral()
2371 CHECK_EQ(src_buf_ptrs.size(), ShapeUtil::TupleElementCount(*shape_)); in MutableBorrowingLiteral()
2375 for (int i = 0; i < src_buf_ptrs.size(); ++i) { in MutableBorrowingLiteral()
2380 child_piece.set_buffer(src_buf_ptrs[i]); in MutableBorrowingLiteral()
2425 BorrowingLiteral::BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs, in BorrowingLiteral() argument
2430 CHECK_EQ(src_buf_ptrs.size(), ShapeUtil::TupleElementCount(*shape_)); in BorrowingLiteral()
2435 for (int i = 0, end = src_buf_ptrs.size(); i < end; ++i) { in BorrowingLiteral()
2438 root_piece_.child(i).set_buffer(const_cast<char*>(src_buf_ptrs[i])); in BorrowingLiteral()
Dliteral.h836 MutableBorrowingLiteral(absl::Span<char*> src_buf_ptrs, const Shape& shape);
874 BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs,
Dliteral_test.cc1693 std::vector<const char*> src_buf_ptrs; in TEST_F() local
1694 src_buf_ptrs.emplace_back( in TEST_F()
1696 src_buf_ptrs.emplace_back(reinterpret_cast<const char*>(hundred.data())); in TEST_F()
1698 src_buf_ptrs, in TEST_F()