| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 26 xla::BorrowingLiteral* literal) { in HostTensorToBorrowingLiteral()
35 xla::BorrowingLiteral* literal) { in HostTensorToBorrowingLiteral()
41 *literal = xla::BorrowingLiteral( in HostTensorToBorrowingLiteral()
47 xla::BorrowingLiteral literal; in HostTensorToLiteral()
70 xla::BorrowingLiteral* literal) { in HostTensorsToBorrowingLiteralTuple()
83 *literal = xla::BorrowingLiteral( in HostTensorsToBorrowingLiteralTuple()
|
| D | literal_util.h | 32 xla::BorrowingLiteral* literal);
38 xla::BorrowingLiteral* literal);
58 xla::BorrowingLiteral* literal);
|
| D | xla_expression.cc | 94 xla::BorrowingLiteral literal; in AsXlaOp()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.h | 78 absl::InlinedVector<BorrowingLiteral, 1> leaves;
153 struct type_caster<xla::BorrowingLiteral> {
155 PYBIND11_TYPE_CASTER(xla::BorrowingLiteral, _("xla::BorrowingLiteral"));
180 value = xla::BorrowingLiteral(buffers,
188 value = xla::BorrowingLiteral(c->buf_ptr, c->shape);
200 type_caster<xla::BorrowingLiteral> literal_caster;
206 value = static_cast<const xla::BorrowingLiteral&>(literal_caster);
|
| D | types.cc | 232 pybind11::detail::type_caster<BorrowingLiteral> caster; in GetPythonBufferTree()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 212 xla::BorrowingLiteral literal; in AutoTransposeAndLinearize()
441 xla::BorrowingLiteral literal; in DoWork()
508 xla::BorrowingLiteral tuple; in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 575 friend class BorrowingLiteral; variable
864 class BorrowingLiteral : public LiteralBase {
866 BorrowingLiteral() : LiteralBase() {} in BorrowingLiteral() function
872 BorrowingLiteral(const char* src_buf_ptr, const Shape& shape);
874 BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs,
|
| D | literal.cc | 2399 void BorrowingLiteral::BuildPieceSubtree(const Shape& shape, Piece* piece) { in BuildPieceSubtree()
2415 BorrowingLiteral::BorrowingLiteral(const char* src_buf_ptr, const Shape& shape) in BorrowingLiteral() function in xla::BorrowingLiteral
2425 BorrowingLiteral::BorrowingLiteral(absl::Span<const char* const> src_buf_ptrs, in BorrowingLiteral() function in xla::BorrowingLiteral
|
| D | literal_test.cc | 1678 BorrowingLiteral literal(reinterpret_cast<const char*>(int64_values.data()), in TEST_F()
1697 auto literal_tuple = BorrowingLiteral( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 158 const ::xla::BorrowingLiteral& literal, int device_ordinal);
|
| D | local_client.cc | 449 const ::xla::BorrowingLiteral& literal, int device_ordinal) { in TransferToLocalServer()
|
| D | xla_builder.h | 2559 BorrowingLiteral literal( in ConstantR1()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 333 xla::BorrowingLiteral literal = in Compute()
334 shape_.IsTuple() ? xla::BorrowingLiteral(tensors_data, shape_) in Compute()
335 : xla::BorrowingLiteral(tensors_data.front(), shape_); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 154 xla::BorrowingLiteral literal( in CopyCPUTensorToDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client_extension.cc | 117 std::vector<BorrowingLiteral> leaves; in PYBIND11_MODULE()
|
| D | tpu_client.h | 188 std::vector<BorrowingLiteral> leaves_literals, const Shape& tuple_shape,
|
| D | tpu_client.cc | 174 std::vector<BorrowingLiteral> leaves, const Shape& tuple_shape, in FromLiterals()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 146 BorrowingLiteral literal(reinterpret_cast<const char*>(&test_vector[1]), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 789 BorrowingLiteral literal(static_cast<const char*>(staging_buffer.get()), in BufferFromHostBuffer()
794 BorrowingLiteral literal(static_cast<const char*>(data), shape); in BufferFromHostBuffer()
|