| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_test.cc | 40 auto on_false = ConstantR0<float>(&builder, 42.0f); in TEST_F() local
41 Select(pred, on_true, on_false); in TEST_F()
50 auto on_false = ConstantR0<int32>(&builder, 42); in TEST_F() local
51 Select(pred, on_true, on_false); in TEST_F()
60 auto on_false = ConstantR0<float>(&builder, 42.0f); in TEST_F() local
61 Select(pred, on_true, on_false); in TEST_F()
70 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
71 Select(pred, on_true, on_false); in XLA_TEST_F()
81 auto on_false = in TEST_F() local
83 Select(pred, on_true, on_false); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 37 llvm::Value* on_true, llvm::Value* on_false, in EmitTupleSelect() argument
60 b->CreateInBoundsGEP(on_false, element_index); in EmitTupleSelect()
|
| D | tuple_ops.h | 63 llvm::Value* on_true, llvm::Value* on_false,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 277 HloInstruction* on_false) { in MakeSelectHlo() argument
280 DCHECK_EQ(computation, on_false->parent()); in MakeSelectHlo()
283 HloOpcode::kSelect, pred, on_true, on_false)); in MakeSelectHlo()
285 select_shape, HloOpcode::kSelect, pred, on_true, on_false)); in MakeSelectHlo()
|
| D | shape_inference.h | 315 const Shape& on_false);
319 const Shape& on_false);
|
| D | hlo_creation_utils.h | 130 HloInstruction* on_false);
|
| D | shape_inference.cc | 2773 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape() argument
2774 if (!ShapeUtil::CompatibleIgnoringFpPrecision(on_true, on_false)) { in InferSelectShape()
2777 ShapeUtil::HumanString(on_true), ShapeUtil::HumanString(on_false)); in InferSelectShape()
2793 on_true, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
2812 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferTupleSelectShape() argument
2815 if (!ShapeUtil::Compatible(on_true, on_false)) { in InferTupleSelectShape()
2818 ShapeUtil::HumanString(on_true), ShapeUtil::HumanString(on_false)); in InferTupleSelectShape()
|
| D | tuple_points_to_analysis.cc | 404 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
406 const PointsToSet& false_points_to_set = *PerInst(on_false)->points_to_set; in HandleTupleSelect()
|
| D | hlo_evaluator.cc | 1291 const auto& on_false = GetEvaluatedLiteralFor(select->operand(2)); in HandleSelect() local
1298 evaluated_[select] = on_false.Clone(); in HandleSelect()
1309 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect() local
1314 evaluated_[tuple_select] = on_false.Clone(); in HandleTupleSelect()
|
| D | hlo_evaluator_test.cc | 226 auto on_false = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P() local
234 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_false))); in TEST_P()
|
| D | hlo_evaluator_typed_visitor.h | 968 [](bool pred, ReturnT on_true, ReturnT on_false) { in HandleSelect()
972 return on_false; in HandleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 430 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
436 GetBasePointer(*on_true), GetBasePointer(*on_false), in HandleTupleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1309 def Select(self, pred, on_true, on_false): argument
1315 return self._client.Select(pred, on_true, on_false)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 370 XlaOp Select(const XlaOp& pred, const XlaOp& on_true, const XlaOp& on_false);
758 const XlaOp& on_false);
1256 XlaOp Select(const XlaOp& pred, const XlaOp& on_true, const XlaOp& on_false);
|
| D | xla_builder.cc | 965 const XlaOp& on_false) { in Select() argument
968 TF_ASSIGN_OR_RETURN(const Shape& false_shape, GetShape(on_false)); in Select()
972 return TernaryOp(opcode, pred, on_true, on_false); in Select()
2909 XlaOp Select(const XlaOp& pred, const XlaOp& on_true, const XlaOp& on_false) { in Select() argument
2910 return pred.builder()->Select(pred, on_true, on_false); in Select()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2225 <b> `Select(pred, on_true, on_false)` </b>
2231 `on_false` | `XlaOp` | array of type T
2233 The arrays `on_true` and `on_false` must have the same shape. This is also the
2235 `on_true` and `on_false`, with the `PRED` element type.
2238 taken from `on_true` if the value of `P` is `true`, and from `on_false` if the
2241 wholly from `on_true` if `pred` is `true`, and from `on_false` if `pred` is `false`.
2264 types for this purpose. If `on_true` and `on_false` are tuples (which must have
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 314 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
321 GetEmittedValueFor(on_false), &b_); in HandleTupleSelect()
|
| /external/python/cpython3/Doc/faq/ |
| D | programming.rst | 722 [on_true] if [expression] else [on_false]
730 [expression] and [on_true] or [on_false]
|
| /external/python/cpython2/Doc/faq/ |
| D | programming.rst | 877 [on_true] if [expression] else [on_false]
|