Searched refs:host_buffer_semantics (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.cc | 104 PjRtClient::HostBufferSemantics host_buffer_semantics) { in PjRtBufferFromPyval() argument
127 if (host_buffer_semantics != in PjRtBufferFromPyval()
141 c->buf_ptr, c->shape, host_buffer_semantics, in PjRtBufferFromPyval()
148 PjRtClient::HostBufferSemantics host_buffer_semantics) { in BufferFromPyval() argument
151 PjRtBufferFromPyval(argument, device, force_copy, host_buffer_semantics)); in BufferFromPyval()
|
| D | py_client.h | 130 PjRtClient::HostBufferSemantics host_buffer_semantics);
133 PjRtClient::HostBufferSemantics host_buffer_semantics);
|
| D | xla_client_test.py | 2051 x, host_buffer_semantics=xla_client.HostBufferSemantics.ZERO_COPY)
2062 x, host_buffer_semantics=during_call)
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 649 HostBufferSemantics host_buffer_semantics, in BufferFromHostBuffer() argument
679 host_buffer_semantics == HostBufferSemantics::kZeroCopy && in BufferFromHostBuffer()
683 (host_buffer_semantics == in BufferFromHostBuffer()
733 if (host_buffer_semantics == HostBufferSemantics::kImmutableOnlyDuringCall || in BufferFromHostBuffer()
747 if (host_buffer_semantics == HostBufferSemantics::kImmutableOnlyDuringCall) { in BufferFromHostBuffer()
770 host_buffer_semantics]() { in BufferFromHostBuffer()
785 if (host_buffer_semantics != in BufferFromHostBuffer()
|
| D | pjrt_client.h | 227 HostBufferSemantics host_buffer_semantics,
|
| D | pjrt_stream_executor_client.h | 189 HostBufferSemantics host_buffer_semantics,
|