experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use BorrowingLiteral to avoid a copy when constructing an R1

Browse Source 
PiperOrigin-RevId: 241258525
This commit is contained in:
A. Unique TensorFlower 2019-03-31 21:48:48 -07:00 committed by TensorFlower Gardener
parent 77b06b0577
commit 7b0fc100ec
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/xla/client/xla_builder.h
View File

@ -1931,7 +1931,11 @@ XlaOp ConstantR0(XlaBuilder* builder, NativeT value) {
template <typename NativeT>
XlaOp ConstantR1(XlaBuilder* builder, absl::Span<const NativeT> values) {
return ConstantLiteral(builder, LiteralUtil::CreateR1<NativeT>(values));
BorrowingLiteral literal(
reinterpret_cast<const char*>(values.begin()),
ShapeUtil::MakeShape(primitive_util::NativeToPrimitiveType<NativeT>(),
{static_cast<int64>(values.size())}));
return ConstantLiteral(builder, literal);
}
template <typename NativeT>