diff options
author | 2018-04-19 07:13:03 -0700 | |
---|---|---|
committer | 2018-04-19 07:15:46 -0700 | |
commit | b4c37a452d2ed1d1c29ceb70127c4ef6434c44ca (patch) | |
tree | 26521713cf735fc0212a1b09da56cfede0650a05 /tensorflow/compiler/xla/service/conditional_simplifier.cc | |
parent | 40f77655affb162d32b7d4861fa68c35fc3d8f7a (diff) |
Teach the conditinal simplifier about sharding.
PiperOrigin-RevId: 193510638
Diffstat (limited to 'tensorflow/compiler/xla/service/conditional_simplifier.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/conditional_simplifier.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/conditional_simplifier.cc b/tensorflow/compiler/xla/service/conditional_simplifier.cc index f35de08085..e560abc87f 100644 --- a/tensorflow/compiler/xla/service/conditional_simplifier.cc +++ b/tensorflow/compiler/xla/service/conditional_simplifier.cc @@ -69,7 +69,7 @@ static StatusOr<bool> TryRemoveConditional(HloInstruction* conditional) { conditional->shape(), {conditional->mutable_operand(2)}, conditional->false_computation())); } - + conditional->SetupDerivedInstruction(call_op); TF_RETURN_IF_ERROR(computation->ReplaceInstruction(conditional, call_op)); TF_RETURN_IF_ERROR(CallInliner::Inline(call_op).status()); |