| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_test.cc | 39 auto on_true = ConstantR0<float>(&builder, 123.0f); in TEST_F() local
41 Select(pred, on_true, on_false); in TEST_F()
49 auto on_true = ConstantR0<int32>(&builder, -42); in TEST_F() local
51 Select(pred, on_true, on_false); in TEST_F()
59 auto on_true = ConstantR0<float>(&builder, 123.0f); in TEST_F() local
61 Select(pred, on_true, on_false); in TEST_F()
69 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
71 Select(pred, on_true, on_false); in XLA_TEST_F()
79 auto on_true = 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
56 b->CreateInBoundsGEP(on_true, 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 | 276 HloInstruction* on_true, in MakeSelectHlo() argument
279 DCHECK_EQ(computation, on_true->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 | 314 const Shape& on_true,
318 const Shape& on_true,
|
| 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()
2787 .IgnoreDynamicDimension()(pred, on_true) || 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()
2830 return on_true; in InferTupleSelectShape()
|
| D | hlo_creation_utils.h | 129 HloInstruction* on_true,
|
| D | tuple_points_to_analysis.cc | 403 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
405 PointsToSet& points_to_set = CreateCopiedPointsToSet(tuple_select, on_true); in HandleTupleSelect()
|
| D | hlo_evaluator.cc | 1290 const auto& on_true = GetEvaluatedLiteralFor(select->operand(1)); in HandleSelect() local
1296 evaluated_[select] = on_true.Clone(); in HandleSelect()
1308 const auto& on_true = GetEvaluatedLiteralFor(tuple_select->operand(1)); in HandleTupleSelect() local
1312 evaluated_[tuple_select] = on_true.Clone(); in HandleTupleSelect()
|
| D | hlo_evaluator_test.cc | 225 auto on_true = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P() local
228 Shape shape = on_true.shape(); in TEST_P()
232 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_true))); in TEST_P()
|
| D | hlo_evaluator_typed_visitor.h | 968 [](bool pred, ReturnT on_true, ReturnT on_false) { in HandleSelect()
970 return on_true; in HandleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 429 auto on_true = tuple_select->operand(1); 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);
757 friend XlaOp Select(const XlaOp& pred, const XlaOp& on_true,
1256 XlaOp Select(const XlaOp& pred, const XlaOp& on_true, const XlaOp& on_false);
|
| D | xla_builder.cc | 964 XlaOp XlaBuilder::Select(const XlaOp& pred, const XlaOp& on_true, in Select() argument
967 TF_ASSIGN_OR_RETURN(const Shape& true_shape, GetShape(on_true)); 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>
2230 `on_true` | `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/python/cpython3/Doc/faq/ |
| D | programming.rst | 722 [on_true] if [expression] else [on_false]
730 [expression] and [on_true] or [on_false]
732 However, this idiom is unsafe, as it can give wrong results when *on_true*
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 313 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
320 GetEmittedValueFor(on_true), in HandleTupleSelect()
|
| /external/python/cpython2/Doc/faq/ |
| D | programming.rst | 877 [on_true] if [expression] else [on_false]
|