| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_test.cc | 41 auto on_false = builder.ConstantR0<float>(42.0f); in TEST_F() local
42 auto result = builder.Select(pred, on_true, on_false); in TEST_F()
51 auto on_false = builder.ConstantR0<int32>(42); in TEST_F() local
52 auto result = builder.Select(pred, on_true, on_false); in TEST_F()
61 auto on_false = builder.ConstantR0<float>(42.0f); in TEST_F() local
62 auto result = builder.Select(pred, on_true, on_false); in TEST_F()
71 auto on_false = builder.ConstantR1<float>({}); in XLA_TEST_F() local
72 auto select = builder.Select(pred, on_true, on_false); in XLA_TEST_F()
81 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 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 | 34 llvm::Value* on_false, llvm::IRBuilder<>* ir_builder, in EmitTupleSelect() argument
58 ir_builder->CreateInBoundsGEP(on_false, element_index); in EmitTupleSelect()
|
| D | tuple_ops.h | 63 llvm::Value* on_false, llvm::IRBuilder<>* ir_builder,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 392 auto on_false = select->operand(2); in HandleSelect() local
394 const PointsToSet& false_points_to_set = *PerInst(on_false)->points_to_set; in HandleSelect()
|
| D | shape_inference.h | 281 const Shape& on_false);
|
| D | shape_inference.cc | 2383 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape() argument
2388 compatible = ShapeUtil::Compatible(on_true, on_false); in InferSelectShape()
2390 compatible = ShapeUtil::CompatibleIgnoringFpPrecision(on_true, on_false); in InferSelectShape()
2396 ShapeUtil::HumanString(on_false).c_str()); in InferSelectShape()
2408 on_true, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
|
| D | hlo_evaluator_test.cc | 160 auto on_false = Literal::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P() local
168 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_false))); in TEST_P()
|
| D | hlo_evaluator.cc | 846 [](bool pred, ReturnT on_true, ReturnT on_false) { in HandleSelect() argument
850 return on_false; in HandleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 743 def Select(self, pred, on_true, on_false): argument
753 _unwrap_data_handle(on_false)))
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 424 auto on_false = select->operand(2); in HandleSelect() local
431 GetBasePointer(*on_false), &ir_builder_, module_); in HandleSelect()
|
| /external/tensorflow/tensorflow/docs_src/performance/xla/ |
| D | operation_semantics.md | 1771 <b> `Select(pred, on_true, on_false)` </b>
1777 `on_false` | `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
1788 `on_false` if `pred` is `false`.
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 | 471 const ComputationDataHandle& on_false) { in Select() argument
472 return TernaryOp(TRIOP_SELECT, pred, on_true, on_false); in Select()
|
| D | computation_builder.h | 302 const ComputationDataHandle& on_false);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 316 auto on_false = select->operand(2); in HandleSelect() local
323 GetEmittedValueFor(on_false), &ir_builder_, module_); in HandleSelect()
|
| /external/python/cpython3/Doc/faq/ |
| D | programming.rst | 717 [on_true] if [expression] else [on_false]
725 [expression] and [on_true] or [on_false]
|
| /external/python/cpython2/Doc/faq/ |
| D | programming.rst | 877 [on_true] if [expression] else [on_false]
|