Searched refs:true_operand (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 346 auto true_operand = Parameter(&inner_builder, 1, r0f32_, "param1"); in XLA_TEST_F() local
348 Conditional(pred_cond, true_operand, CreateR0CeilComputation(), false_operand, in XLA_TEST_F()
638 auto true_operand = GetTupleElement(param0, 1); in XLA_TEST_F() local
640 Conditional(pred_cond, true_operand, CreateR0CeilComputation(), in XLA_TEST_F()
668 auto true_operand = GetTupleElement(param0, 1); in XLA_TEST_F() local
670 Conditional(pred_cond, true_operand, CreateR0CeilComputation(), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1747 true_operand = ops.Constant(c, np.float32(3.))
1752 ops.Conditional(pred, true_operand, true_computation, false_operand,
1759 true_operand = ops.Constant(c, np.float32(3.))
1764 ops.Conditional(pred, true_operand, true_computation, false_operand,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 831 XlaOp Conditional(XlaOp predicate, XlaOp true_operand,
1378 friend XlaOp Conditional(XlaOp predicate, XlaOp true_operand,
2423 XlaOp Conditional(XlaOp predicate, XlaOp true_operand,
|
| D | xla_builder.cc | 2341 XlaOp XlaBuilder::Conditional(XlaOp predicate, XlaOp true_operand, in Conditional() argument
2357 {true_operand, false_operand}); in Conditional()
4853 XlaOp Conditional(const XlaOp predicate, const XlaOp true_operand, in Conditional() argument
4857 return predicate.builder()->Conditional(predicate, true_operand, in Conditional()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 657 <b> `Conditional(pred, true_operand, true_computation, false_operand,
665 `true_operand` | `XlaOp` | Argument of type \\(T_0\\)
674 be invoked with `true_operand` which must be of the same type. The
|