Searched refs:pred_cond (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 44 llvm::Value* pred_cond = b->CreateICmpNE( in EmitTupleSelect() local
51 VLOG(2) << " pred_cond: " << DumpToString(*pred_cond); in EmitTupleSelect()
66 b->CreateStore(b->CreateSelect(pred_cond, on_true_element, on_false_element, in EmitTupleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 345 auto pred_cond = Parameter(&inner_builder, 0, r0bool, "param0"); in XLA_TEST_F() local
348 Conditional(pred_cond, true_operand, CreateR0CeilComputation(), false_operand, in XLA_TEST_F()
637 auto pred_cond = GetTupleElement(param0, 0); in XLA_TEST_F() local
640 Conditional(pred_cond, true_operand, CreateR0CeilComputation(), in XLA_TEST_F()
667 auto pred_cond = GetTupleElement(param0, 0); in XLA_TEST_F() local
670 Conditional(pred_cond, true_operand, CreateR0CeilComputation(), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis_test.cc | 1815 auto pred_cond = inner_builder.AddInstruction( in TEST_P() local
1823 scalar_shape_, pred_cond, true_operand_cond, computation1, in TEST_P()
1874 EXPECT_FALSE(analysis.ValueIsDefinedAt(pred_cond)); in TEST_P()
1879 EXPECT_EQ(analysis.GetUniqueValueAt(pred_cond), in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2560 llvm::Value* pred_cond = in HandleConditional() local
2566 llvm_ir::EmitIfThenElse(pred_cond, "conditional", &b_); in HandleConditional()
|