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

Automated g4 rollback of changelist 193457083

Browse Source 
PiperOrigin-RevId: 200275448
This commit is contained in:
A. Unique TensorFlower 2018-06-12 13:57:52 -07:00 committed by TensorFlower Gardener
parent 2d8b5115ab
commit 9c7ba75034
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/compiler/xla/service/reshape_mover.cc
View File

@ -155,20 +155,15 @@ HloInstruction* UpdateOperand(const HloInstruction* first_reshape_operand,
case HloOpcode::kConstant: {
if (first_reshape_operand->opcode() == HloOpcode::kReshape) {
VLOG(5) << "Adding reshape to kConstant operand";
HloInstruction* reshape = computation->AddInstruction(
return computation->AddInstruction(
HloInstruction::CreateReshape(new_shape, operand));
operand->SetupDerivedInstruction(reshape);
return reshape;
} else {
CHECK(first_reshape_operand->opcode() == HloOpcode::kTranspose);
VLOG(5) << "Adding transpose to kConstant operand";
std::vector<int64> inverse_permutation =
InversePermutation(first_reshape_operand->dimensions());
HloInstruction* transpose =
computation->AddInstruction(HloInstruction::CreateTranspose(
new_shape, operand, inverse_permutation));
operand->SetupDerivedInstruction(transpose);
return transpose;
return computation->AddInstruction(HloInstruction::CreateTranspose(
new_shape, operand, inverse_permutation));
}
}
case HloOpcode::kRng: {