Searched refs:BorrowingLiteral (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 26 xla::BorrowingLiteral* literal) { in HostTensorToBorrowingLiteral()
30 *literal = xla::BorrowingLiteral( in HostTensorToBorrowingLiteral()
36 xla::BorrowingLiteral literal; in HostTensorToLiteral()
59 xla::BorrowingLiteral* literal) { in HostTensorsToBorrowingLiteralTuple()
72 *literal = xla::BorrowingLiteral( in HostTensorsToBorrowingLiteralTuple()
|
| D | literal_util.h | 32 xla::BorrowingLiteral* literal);
52 xla::BorrowingLiteral* literal);
|
| D | xla_expression.cc | 84 xla::BorrowingLiteral literal; in AsXlaOp()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 541 friend class BorrowingLiteral; variable
839 class BorrowingLiteral : public LiteralBase {
841 BorrowingLiteral() : LiteralBase() {} in BorrowingLiteral() function
847 BorrowingLiteral(const char* src_buf_ptr, const Shape& shape);
849 BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs,
|
| D | literal.cc | 2244 void BorrowingLiteral::BuildPieceSubtree(const Shape& shape, Piece* piece) { in BuildPieceSubtree()
2260 BorrowingLiteral::BorrowingLiteral(const char* src_buf_ptr, const Shape& shape) in BorrowingLiteral() function in xla::BorrowingLiteral
2270 BorrowingLiteral::BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs, in BorrowingLiteral() function in xla::BorrowingLiteral
|
| D | literal_test.cc | 1609 BorrowingLiteral literal(reinterpret_cast<const char*>(int64_values.data()), in TEST_F()
1628 auto literal_tuple = BorrowingLiteral( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 134 const ::xla::BorrowingLiteral& literal, int device_oridinal);
|
| D | local_client.cc | 314 const ::xla::BorrowingLiteral& literal, int device_oridinal) { in TransferToLocalServer()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 283 xla::BorrowingLiteral literal = in Compute()
284 shape_.IsTuple() ? xla::BorrowingLiteral(tensors_data, shape_) in Compute()
285 : xla::BorrowingLiteral(tensors_data.front(), shape_); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 146 xla::BorrowingLiteral literal( in CopyCPUTensorToDevice()
|