Searched refs:true_operand (Results 1 – 8 of 8) 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 | 1630 true_operand = c.ConstantF32Scalar(3.)
1634 c.Conditional(pred, true_operand, true_computation, false_operand,
1641 true_operand = c.ConstantF32Scalar(3.)
1645 c.Conditional(pred, true_operand, true_computation, false_operand,
|
| D | local_computation_builder.cc | 695 const LocalOp& true_operand, in Conditional() argument
699 return xla::Conditional(predicate.op(), true_operand.op(), in Conditional()
|
| D | local_computation_builder.h | 347 LocalOp Conditional(const LocalOp& predicate, const LocalOp& true_operand,
|
| D | xla_client.py | 1557 def Conditional(self, pred, true_operand, true_computation, false_operand, argument
1569 return self._client.Conditional(pred, true_operand,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 531 XlaOp Conditional(const XlaOp& predicate, const XlaOp& true_operand,
958 friend XlaOp Conditional(const XlaOp& predicate, const XlaOp& true_operand,
1796 XlaOp Conditional(const XlaOp& predicate, const XlaOp& true_operand,
|
| D | xla_builder.cc | 1891 XlaOp XlaBuilder::Conditional(const XlaOp& predicate, const XlaOp& true_operand, in Conditional() argument
1898 {true_operand, false_operand}); in Conditional()
3402 XlaOp Conditional(const XlaOp& predicate, const XlaOp& true_operand, in Conditional() argument
3406 return predicate.builder()->Conditional(predicate, true_operand, in Conditional()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 540 <b> `Conditional(pred, true_operand, true_computation, false_operand,
546 `true_operand` | `XlaOp` | Argument of type $$ T_0 $$
555 be invoked with `true_operand` which must be of the same type. The
|