/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]
|