OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[XLA] Support conditional in all backends.

Browse Source 
PiperOrigin-RevId: 179900775
This commit is contained in:
A. Unique TensorFlower
2017-12-22 00:07:40 -08:00
committed by TensorFlower Gardener
parent a2039aa91f
commit 531bf79775
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/xla/service/hlo_instruction.cc
View File

@@ -1684,9 +1684,11 @@ bool HloInstruction::IdenticalSlowPath(
return custom_call_target_ == other.custom_call_target_;
case HloOpcode::kReverse:
return dimensions() == other.dimensions();
case HloOpcode::kConditional:
return eq_computations(true_computation(), other.true_computation()) &&
eq_computations(false_computation(), other.false_computation());
// These opcodes are not yet supported.
case HloOpcode::kConditional:
case HloOpcode::kInfeed:
case HloOpcode::kOutfeed:
case HloOpcode::kSort:

6
tensorflow/compiler/xla/tests/BUILD
View File

@@ -442,12 +442,6 @@ xla_test(
xla_test(
name = "conditional_test",
srcs = ["conditional_test.cc"],
# Currently, Conditional is supported only in CPU and GPU backends.
backends = [
"cpu",
"gpu",
"cpu_parallel",
],
deps = [
"//tensorflow/compiler/xla:xla_data_proto",
"//tensorflow/compiler/xla/client:computation_builder",