Searched refs:ToLiteral (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 307 virtual void ToLiteral(MutableLiteralBase* literal,
311 Status ToLiteral(MutableLiteralBase* literal) { in ToLiteral() function
314 ToLiteral(literal, [&](Status s) { in ToLiteral()
324 StatusOr<std::shared_ptr<Literal>> ToLiteral() { in ToLiteral() function
327 TF_RETURN_IF_ERROR(ToLiteral(literal.get())); in ToLiteral()
|
| D | gpu_multistream_test.cc | 98 TF_ASSERT_OK_AND_ASSIGN(auto out_literal, out_buffers[0][0]->ToLiteral()); in TEST()
100 TF_ASSERT_OK_AND_ASSIGN(out_literal, out_buffers[0][1]->ToLiteral()); in TEST()
|
| D | pjrt_stream_executor_client.h | 487 using PjRtBuffer::ToLiteral;
488 void ToLiteral(MutableLiteralBase* literal,
|
| D | pjrt_stream_executor_client.cc | 1212 void PjRtStreamExecutorBuffer::ToLiteral(MutableLiteralBase* literal, in ToLiteral() function in xla::PjRtStreamExecutorBuffer
1365 TF_ASSIGN_OR_RETURN(std::shared_ptr<Literal> literal, ToLiteral()); in CopyToDevice()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.h | 145 Status ToLiteral(xla::Backend* backend, xla::MutableLiteralBase* literal);
|
| D | xrt_state.cc | 242 Status XRTTupleAllocation::ToLiteral(xla::Backend* backend, in ToLiteral() function in tensorflow::XRTTupleAllocation
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client_extension.cc | 167 TF_ASSIGN_OR_RETURN(literal, buffer->ToLiteral()); in PYBIND11_MODULE()
|
| D | tpu_client.h | 217 StatusOr<std::shared_ptr<Literal>> ToLiteral();
|
| D | tpu_client.cc | 344 StatusOr<std::shared_ptr<Literal>> PyTpuBuffer::ToLiteral() { in ToLiteral() function in xla::PyTpuBuffer
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_buffer.cc | 99 buffer_->ToLiteral(literal, in CopyToHostAsync()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 522 OP_REQUIRES_OK(ctx, allocation->ToLiteral(device_ref.backend(), &literal)); in Compute()
608 TF_RETURN_IF_ERROR(sub->ToLiteral(device_ref.backend(), &literal)); in Compute()
|