| /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/ |
| D | conditional_code_motion_test.cc | 283 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
284 ASSERT_EQ(on_true->instruction_count(), 1); in TEST_F()
351 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
352 ASSERT_EQ(on_true->instruction_count(), 9); in TEST_F()
406 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
407 ASSERT_EQ(on_true->instruction_count(), 1); in TEST_F()
608 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
609 ASSERT_EQ(on_true->instruction_count(), 5); in TEST_F()
695 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
696 ASSERT_EQ(on_true->instruction_count(), 6); in TEST_F()
[all …]
|
| D | hlo_creation_utils.cc | 382 HloInstruction* on_true, in MakeSelectHlo() argument
386 DCHECK_EQ(computation, on_true->parent()); in MakeSelectHlo()
388 Shape op_shape = on_true->shape(); in MakeSelectHlo()
406 on_true, on_false)); in MakeSelectHlo()
409 select_shape, select_op_code, pred, on_true, on_false)); in MakeSelectHlo()
|
| D | shape_inference.h | 360 const Shape& on_true,
364 const Shape& on_true,
|
| D | shape_inference.cc | 3127 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape() argument
3129 TF_RETURN_IF_ERROR(ExpectArray(on_true, "select on-true")); in InferSelectShape()
3132 if (!ShapeUtil::CompatibleIgnoringFpPrecision(on_true, on_false)) { in InferSelectShape()
3135 ShapeUtil::HumanString(on_true), ShapeUtil::HumanString(on_false)); in InferSelectShape()
3145 .IgnoreDynamicDimension()(pred, on_true)) { in InferSelectShape()
3149 ShapeUtil::HumanString(on_true), ShapeUtil::HumanString(pred)); 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()
[all …]
|
| D | hlo_creation_utils.h | 171 HloInstruction* on_true,
|
| D | tuple_points_to_analysis.cc | 453 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
455 PointsToSet& points_to_set = CreateCopiedPointsToSet(tuple_select, on_true); in HandleTupleSelect()
|
| D | hlo_evaluator.cc | 2002 const auto& on_true = GetEvaluatedLiteralFor(select->operand(1)); in HandleSelect() local
2008 evaluated_[select] = on_true.Clone(); in HandleSelect()
2020 const auto& on_true = GetEvaluatedLiteralFor(tuple_select->operand(1)); in HandleTupleSelect() local
2024 evaluated_[tuple_select] = on_true.Clone(); in HandleTupleSelect()
|
| D | hlo_evaluator_test.cc | 295 auto on_true = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P() local
298 Shape shape = on_true.shape(); in TEST_P()
302 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_true))); in TEST_P()
|
| D | hlo_evaluator_typed_visitor.h | 1045 [](bool pred, ReturnT on_true, ReturnT on_false) {
1047 return on_true;
|
| /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/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1805 if (on_true() == on_false()) { in fold()
1806 return on_true(); in fold()
1820 return predicate.getSplatValue<APInt>().getBoolValue() ? on_true() in fold()
|
| /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
1227 TF_ASSIGN_OR_RETURN(const Shape* true_shape, GetShapePtr(on_true)); 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 | 483 Arg<LHLO_Buffer, "", [MemRead]>:$on_true,
|
| D | hlo_ops.td | 1113 HLO_Tensor:$on_true,
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2511 <b> `Select(pred, on_true, on_false)` </b>
2516 `on_true` | `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 | 2615 Value on_true = broadcast_or_self(op.t()); in matchAndRewrite() local
2618 rewriter.replaceOpWithNewOp<SelectOp>(op, on_true.getType(), pred, on_true, in matchAndRewrite()
|
| /external/python/cpython3/Doc/faq/ |
| D | programming.rst | 714 [on_true] if [expression] else [on_false]
722 [expression] and [on_true] or [on_false]
724 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 | 377 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
384 GetEmittedValueFor(on_true), in HandleTupleSelect()
|
| /external/python/cpython2/Doc/faq/ |
| D | programming.rst | 877 [on_true] if [expression] else [on_false]
|