Searched refs:true_operand (Results 1 – 6 of 6) 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.py | 1337 def Conditional(self, pred, true_operand, true_computation, false_operand, argument
1349 return ops.Conditional(pred, true_operand, true_computation.computation,
|
| D | xla_client_test.py | 1902 true_operand = c.ConstantF32Scalar(3.)
1906 c.Conditional(pred, true_operand, true_computation, false_operand,
1913 true_operand = c.ConstantF32Scalar(3.)
1917 c.Conditional(pred, true_operand, true_computation, false_operand,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 571 XlaOp Conditional(XlaOp predicate, XlaOp true_operand,
990 friend XlaOp Conditional(XlaOp predicate, XlaOp true_operand,
1864 XlaOp Conditional(XlaOp predicate, XlaOp true_operand,
|
| D | xla_builder.cc | 1832 XlaOp XlaBuilder::Conditional(XlaOp predicate, XlaOp true_operand, in Conditional() argument
1848 {true_operand, false_operand}); in Conditional()
3468 XlaOp Conditional(const XlaOp predicate, const XlaOp true_operand, in Conditional() argument
3472 return predicate.builder()->Conditional(predicate, true_operand, in Conditional()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 614 <b> `Conditional(pred, true_operand, true_computation, false_operand,
620 `true_operand` | `XlaOp` | Argument of type $$ T_0 $$
629 be invoked with `true_operand` which must be of the same type. The
|