Searched refs:scoped_buffer (Results 1 – 4 of 4) sorted by relevance
268 TF_ASSIGN_OR_RETURN(auto scoped_buffer, in LiteralToShapedBuffer()274 stream.get(), literal, scoped_buffer)); in LiteralToShapedBuffer()275 return std::move(scoped_buffer); in LiteralToShapedBuffer()
40 auto scoped_buffer = absl::make_unique<xla::ScopedShapedBuffer>( in TEST() local42 std::unique_ptr<xla::ShapedBuffer> buffer = std::move(scoped_buffer); in TEST()
189 std::unique_ptr<xla::ScopedShapedBuffer> scoped_buffer; in CreateAndTransfer() local191 backend, device_ordinal, literal.shape(), &scoped_buffer)); in CreateAndTransfer()194 stream.get(), literal, *scoped_buffer)); in CreateAndTransfer()200 auto shaped_buffer = scoped_buffer->release(); in CreateAndTransfer()
225 auto scoped_buffer = run_result.ConsumeValueOrDie(); in DoWork() local226 auto shaped_buffer = scoped_buffer.release(); in DoWork()