Searched defs:on_true (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_test.cc | 39 auto on_true = ConstantR0<float>(&builder, 123.0f); in TEST_F() local
49 auto on_true = ConstantR0<int32>(&builder, -42); in TEST_F() local
59 auto on_true = ConstantR0<float>(&builder, 123.0f); in TEST_F() local
69 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
79 auto on_true = in TEST_F() local
96 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
110 auto on_true = in TEST_F() local
126 auto on_true = in TEST_F() local
205 auto on_true = ConstantR1<float>(&builder, {11.0f, 22.0f, 33.0f, 44.0f}); in TEST_F() local
222 auto on_true = ConstantR1<float>(&builder, {11.0f, 22.0f, 33.0f, 44.0f}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 37 llvm::Value* on_true, llvm::Value* on_false, in EmitTupleSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 323 HloInstruction* on_true, in MakeSelectHlo()
|
| D | tuple_points_to_analysis.cc | 451 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
|
| D | shape_inference.cc | 2984 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape()
3014 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferTupleSelectShape()
|
| D | hlo_evaluator.cc | 1990 const auto& on_true = GetEvaluatedLiteralFor(select->operand(1)); in HandleSelect() local
2008 const auto& on_true = GetEvaluatedLiteralFor(tuple_select->operand(1)); in HandleTupleSelect() local
|
| D | hlo_evaluator_test.cc | 294 auto on_true = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 466 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1084 XlaOp XlaBuilder::Select(XlaOp pred, XlaOp on_true, XlaOp on_false) { in Select()
2974 XlaOp Select(const XlaOp pred, const XlaOp on_true, const XlaOp on_false) { in Select()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 372 auto on_true = tuple_select->operand(1); in HandleTupleSelect() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1342 Value on_true = broadcast_or_self(op.t()); in matchAndRewrite() local
|