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

[XLA] Keep op metadata when replacing with bitcast.

Browse Source 
PiperOrigin-RevId: 316667446
Change-Id: I3618b5eafba269eb087ff7607bdad1cb3873ffbc
This commit is contained in:
Thomas Joerg 2020-06-16 06:21:33 -07:00 committed by TensorFlower Gardener
parent 35fc231281
commit b8b6bc58a1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/xla/service/algebraic_simplifier.cc
View File

@ -573,6 +573,7 @@ void AlgebraicSimplifierVisitor::ReplaceWithBitcast(HloInstruction* instruction,
auto bitcast = computation_->AddInstruction( auto bitcast = computation_->AddInstruction(
HloInstruction::CreateBitcast(instruction->shape(), operand)); HloInstruction::CreateBitcast(instruction->shape(), operand));
bitcast->set_metadata(instruction->metadata());
TF_CHECK_OK(ReplaceInstruction(instruction, bitcast)); TF_CHECK_OK(ReplaceInstruction(instruction, bitcast));
} }