Searched refs:HostBufferSemantics (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_multistream_test.cc | 76 PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes, in TEST()
82 PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes, in TEST()
88 PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes, in TEST()
|
| D | pjrt_client.h | 198 enum class HostBufferSemantics { enum
227 HostBufferSemantics host_buffer_semantics,
|
| D | pjrt_stream_executor_client.cc | 649 HostBufferSemantics host_buffer_semantics, in BufferFromHostBuffer()
679 host_buffer_semantics == HostBufferSemantics::kZeroCopy && in BufferFromHostBuffer()
684 HostBufferSemantics::kImmutableOnlyDuringCall || in BufferFromHostBuffer()
733 if (host_buffer_semantics == HostBufferSemantics::kImmutableOnlyDuringCall || in BufferFromHostBuffer()
747 if (host_buffer_semantics == HostBufferSemantics::kImmutableOnlyDuringCall) { in BufferFromHostBuffer()
786 HostBufferSemantics::kImmutableOnlyDuringCall) { in BufferFromHostBuffer()
1370 PjRtStreamExecutorClient::HostBufferSemantics::kZeroCopy, in CopyToDevice()
|
| D | pjrt_stream_executor_client.h | 189 HostBufferSemantics host_buffer_semantics,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 196 py::enum_<PjRtClient::HostBufferSemantics>(m, "HostBufferSemantics") in PYBIND11_MODULE()
198 PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall) in PYBIND11_MODULE()
200 PjRtClient::HostBufferSemantics::kImmutableUntilTransferCompletes) in PYBIND11_MODULE()
201 .value("ZERO_COPY", PjRtClient::HostBufferSemantics::kZeroCopy); in PYBIND11_MODULE()
225 PjRtClient::HostBufferSemantics::kZeroCopy) in PYBIND11_MODULE()
|
| D | py_client.h | 130 PjRtClient::HostBufferSemantics host_buffer_semantics);
133 PjRtClient::HostBufferSemantics host_buffer_semantics);
|
| D | py_client.cc | 104 PjRtClient::HostBufferSemantics host_buffer_semantics) { in PjRtBufferFromPyval()
128 PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall) { in PjRtBufferFromPyval()
148 PjRtClient::HostBufferSemantics host_buffer_semantics) { in BufferFromPyval()
|
| D | jax_jit.cc | 369 xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, nullptr, in ConvertToScalarBuffer()
625 xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, in HandleComplex()
634 xla::PjRtClient::HostBufferSemantics::kImmutableOnlyDuringCall, in HandleComplex()
670 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()
|
| D | xla_client.py | 317 HostBufferSemantics = _xla.HostBufferSemantics variable
|
| D | xla_client_test.py | 2051 x, host_buffer_semantics=xla_client.HostBufferSemantics.ZERO_COPY)
2060 during_call = xla_client.HostBufferSemantics.IMMUTABLE_ONLY_DURING_CALL
|