Searched refs:scoped_buffer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 174 std::unique_ptr<xla::ScopedShapedBuffer> scoped_buffer; in CreateAndTransfer() local
177 &scoped_buffer)); in CreateAndTransfer()
180 stream.get(), literal, *scoped_buffer)); in CreateAndTransfer()
186 auto shaped_buffer = scoped_buffer->release(); in CreateAndTransfer()
201 std::unique_ptr<xla::ScopedShapedBuffer> scoped_buffer; in CreateUninitialized() local
203 memory_manager, backend, device_ordinal, shape, &scoped_buffer)); in CreateUninitialized()
209 auto shaped_buffer = scoped_buffer->release(); in CreateUninitialized()
306 std::unique_ptr<xla::ScopedShapedBuffer> scoped_buffer; in SwapIn() local
309 on_host_shape(), &scoped_buffer)); in SwapIn()
315 stream.get(), *literal_, *scoped_buffer)); in SwapIn()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shaped_buffer_test.cc | 38 auto scoped_buffer = absl::make_unique<xla::ScopedShapedBuffer>( in TEST() local
40 std::unique_ptr<xla::ShapedBuffer> buffer = std::move(scoped_buffer); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 405 TF_ASSIGN_OR_RETURN(auto scoped_buffer, in LiteralToShapedBuffer()
411 stream.get(), literal, scoped_buffer)); in LiteralToShapedBuffer()
412 return std::move(scoped_buffer); in LiteralToShapedBuffer()
|