Home
last modified time | relevance | path

Searched refs:kZeroCopy (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc201 .value("ZERO_COPY", PjRtClient::HostBufferSemantics::kZeroCopy); in PYBIND11_MODULE()
225 PjRtClient::HostBufferSemantics::kZeroCopy) in PYBIND11_MODULE()
Djax_jit.cc670 xla::PjRtClient::HostBufferSemantics::kZeroCopy)); in HandleBufferFromPyval()
694 xla::PjRtClient::HostBufferSemantics::kZeroCopy)); in HandleUint64()
705 xla::PjRtClient::HostBufferSemantics::kZeroCopy)); in HandleUint64()
734 xla::PjRtClient::HostBufferSemantics::kZeroCopy)); in HandleNdarray()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h221 kZeroCopy, enumerator
Dpjrt_stream_executor_client.cc679 host_buffer_semantics == HostBufferSemantics::kZeroCopy && in BufferFromHostBuffer()
1370 PjRtStreamExecutorClient::HostBufferSemantics::kZeroCopy, in CopyToDevice()