| /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/ |
| D | conditional_code_motion_test.cc | 285 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
286 ASSERT_EQ(on_false->instruction_count(), 1); in TEST_F()
353 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
354 ASSERT_EQ(on_false->instruction_count(), 9); in TEST_F()
408 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
409 ASSERT_EQ(on_false->instruction_count(), 3); in TEST_F()
610 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
611 ASSERT_EQ(on_false->instruction_count(), 5); in TEST_F()
697 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
698 ASSERT_EQ(on_false->instruction_count(), 6); in TEST_F()
[all …]
|
| D | shape_inference.h | 361 const Shape& on_false);
365 const Shape& on_false);
|
| D | hlo_creation_utils.cc | 383 HloInstruction* on_false, in MakeSelectHlo() argument
387 DCHECK_EQ(computation, on_false->parent()); in MakeSelectHlo()
406 on_true, on_false)); in MakeSelectHlo()
409 select_shape, select_op_code, pred, on_true, on_false)); in MakeSelectHlo()
|
| D | hlo_creation_utils.h | 172 HloInstruction* on_false,
|
| D | shape_inference.cc | 3127 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape() argument
3130 TF_RETURN_IF_ERROR(ExpectArray(on_false, "select on-false")); in InferSelectShape()
3132 if (!ShapeUtil::CompatibleIgnoringFpPrecision(on_true, on_false)) { in InferSelectShape()
3135 ShapeUtil::HumanString(on_true), ShapeUtil::HumanString(on_false)); in InferSelectShape()
3153 pred, ShapeUtil::HigherPrecisionElementType(on_true, on_false)); in InferSelectShape()
3157 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferTupleSelectShape() argument
3160 if (!ShapeUtil::Compatible(on_true, on_false)) { in InferTupleSelectShape()
3163 ShapeUtil::HumanString(on_true), ShapeUtil::HumanString(on_false)); in InferTupleSelectShape()
|
| D | tuple_points_to_analysis.cc | 454 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
456 const PointsToSet& false_points_to_set = *PerInst(on_false)->points_to_set; in HandleTupleSelect()
|
| D | hlo_evaluator.cc | 2003 const auto& on_false = GetEvaluatedLiteralFor(select->operand(2)); in HandleSelect() local
2010 evaluated_[select] = on_false.Clone(); in HandleSelect()
2021 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect() local
2026 evaluated_[tuple_select] = on_false.Clone(); in HandleTupleSelect()
|
| D | hlo_evaluator_typed_visitor.h | 1045 [](bool pred, ReturnT on_true, ReturnT on_false) {
1049 return on_false;
|
| D | hlo_evaluator_test.cc | 296 auto on_false = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P() local
304 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_false))); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 37 llvm::Value* on_true, llvm::Value* on_false, in EmitTupleSelect() argument
53 llvm::Value* src = b->CreateSelect(pred_cond, on_true, on_false); in EmitTupleSelect()
|
| D | tuple_ops.h | 63 llvm::Value* on_true, llvm::Value* on_false,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 513 XlaOp Select(XlaOp pred, XlaOp on_true, XlaOp on_false);
1109 friend XlaOp Select(XlaOp pred, XlaOp on_true, XlaOp on_false);
1802 XlaOp Select(XlaOp pred, XlaOp on_true, XlaOp on_false);
|
| D | xla_builder.cc | 1225 XlaOp XlaBuilder::Select(XlaOp pred, XlaOp on_true, XlaOp on_false) { in Select() argument
1228 TF_ASSIGN_OR_RETURN(const Shape* false_shape, GetShapePtr(on_false)); in Select()
1232 return TernaryOp(opcode, pred, on_true, on_false); in Select()
4186 XlaOp Select(const XlaOp pred, const XlaOp on_true, const XlaOp on_false) { in Select() argument
4187 return pred.builder()->Select(pred, on_true, on_false); in Select()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.td | 1164 Constructs an output tensor from the elements of `on_true` and `on_false`
1167 `pred`, `on_true` and `on_false` must be broadcast compatible.
|
| D | lhlo_ops.td | 484 Arg<LHLO_Buffer, "", [MemRead]>:$on_false,
|
| D | hlo_ops.td | 1114 HLO_Tensor:$on_false
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2511 <b> `Select(pred, on_true, on_false)` </b>
2517 `on_false` | `XlaOp` | array of type T
2519 The arrays `on_true` and `on_false` must have the same shape. This is also the
2521 `on_true` and `on_false`, with the `PRED` element type.
2524 taken from `on_true` if the value of `P` is `true`, and from `on_false` if the
2527 wholly from `on_true` if `pred` is `true`, and from `on_false` if `pred` is `false`.
2550 types for this purpose. If `on_true` and `on_false` are tuples (which must have
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 514 // $gradients needs to be of static shape so that on_true and on_false operands
|
| D | legalize_tf.cc | 2616 Value on_false = broadcast_or_self(op.e()); in matchAndRewrite() local
2619 on_false); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1805 if (on_true() == on_false()) { in fold()
1821 : on_false(); in fold()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 378 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
385 GetEmittedValueFor(on_false), &b_); in HandleTupleSelect()
|
| /external/python/cpython3/Doc/faq/ |
| D | programming.rst | 714 [on_true] if [expression] else [on_false]
722 [expression] and [on_true] or [on_false]
|
| /external/python/cpython2/Doc/faq/ |
| D | programming.rst | 877 [on_true] if [expression] else [on_false]
|