| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_test.cc | 40 auto on_true = builder.ConstantR0<float>(123.0f); in TEST_F() local
42 auto result = builder.Select(pred, on_true, on_false); in TEST_F()
50 auto on_true = builder.ConstantR0<int32>(-42); in TEST_F() local
52 auto result = builder.Select(pred, on_true, on_false); in TEST_F()
60 auto on_true = builder.ConstantR0<float>(123.0f); in TEST_F() local
62 auto result = builder.Select(pred, on_true, on_false); in TEST_F()
70 auto on_true = builder.ConstantR1<float>({}); in XLA_TEST_F() local
72 auto select = builder.Select(pred, on_true, on_false); in XLA_TEST_F()
80 auto on_true = builder.ConstantR1<float>({-2.5f, 25.5f, 2.25f, -10.0f, 6.0f}); in TEST_F() local
82 auto select = builder.Select(pred, on_true, on_false); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 33 void EmitTupleSelect(IrArray select, IrArray pred, llvm::Value* on_true, in EmitTupleSelect() argument
53 ir_builder->CreateInBoundsGEP(on_true, element_index); in EmitTupleSelect()
|
| D | tuple_ops.h | 62 void EmitTupleSelect(IrArray select, IrArray pred, llvm::Value* on_true,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 2383 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape() argument
2385 if (ShapeUtil::IsTuple(on_true)) { in InferSelectShape()
2388 compatible = ShapeUtil::Compatible(on_true, on_false); in InferSelectShape()
2390 compatible = ShapeUtil::CompatibleIgnoringFpPrecision(on_true, on_false); in InferSelectShape()
2395 ShapeUtil::HumanString(on_true).c_str(), in InferSelectShape()
2403 if (ShapeUtil::SameDimensions(pred, on_true) || ShapeUtil::Rank(pred) == 0) { in InferSelectShape()
2408 on_true, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
|
| D | tuple_points_to_analysis.cc | 391 auto on_true = select->operand(1); in HandleSelect() local
393 PointsToSet& points_to_set = CreateCopiedPointsToSet(select, on_true); in HandleSelect()
|
| D | shape_inference.h | 280 const Shape& on_true,
|
| D | hlo_evaluator_test.cc | 159 auto on_true = Literal::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P() local
162 Shape shape = on_true->shape(); in TEST_P()
166 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_true))); in TEST_P()
|
| D | hlo_evaluator.cc | 846 [](bool pred, ReturnT on_true, ReturnT on_false) { in HandleSelect() argument
848 return on_true; in HandleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 743 def Select(self, pred, on_true, on_false): argument
752 _unwrap_data_handle(on_true),
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 423 auto on_true = select->operand(1); in HandleSelect() local
430 GetBasePointer(*on_true), in HandleSelect()
|
| /external/tensorflow/tensorflow/docs_src/performance/xla/ |
| D | operation_semantics.md | 1771 <b> `Select(pred, on_true, on_false)` </b>
1776 `on_true` | `ComputationDataHandle` | array of type T
1779 The arrays `on_true` and `on_false` must have the same shape. This is also the
1781 `on_true` and `on_false`, with the `PRED` element type.
1784 taken from `on_true` if the value of `P` is `true`, and from `on_false` if the
1787 output array is taken wholly from `on_true` if `pred` is `true`, and from
1811 types for this purpose. If `on_true` and `on_false` are tuples (which must have
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | computation_builder.cc | 470 const ComputationDataHandle& pred, const ComputationDataHandle& on_true, in Select() argument
472 return TernaryOp(TRIOP_SELECT, pred, on_true, on_false); in Select()
|
| D | computation_builder.h | 301 const ComputationDataHandle& on_true,
|
| /external/python/cpython3/Doc/faq/ |
| D | programming.rst | 717 [on_true] if [expression] else [on_false]
725 [expression] and [on_true] or [on_false]
727 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 | 315 auto on_true = select->operand(1); in HandleSelect() local
322 GetIrArrayFor(select), GetIrArrayFor(pred), GetEmittedValueFor(on_true), in HandleSelect()
|
| /external/python/cpython2/Doc/faq/ |
| D | programming.rst | 877 [on_true] if [expression] else [on_false]
|