| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_test.cc | 40 auto on_false = ConstantR0<float>(&builder, 42.0f); in TEST_F() local
50 auto on_false = ConstantR0<int32>(&builder, 42); in TEST_F() local
60 auto on_false = ConstantR0<float>(&builder, 42.0f); in TEST_F() local
70 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
81 auto on_false = in TEST_F() local
97 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
112 auto on_false = in TEST_F() local
128 auto on_false = in TEST_F() local
206 auto on_false = in TEST_F() local
223 auto on_false = in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_code_motion_test.cc | 286 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
354 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
409 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
611 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
698 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
747 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
824 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
863 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
904 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
938 const HloComputation* on_false = conditional->branch_computation(1); in TEST_F() local
[all …]
|
| D | hlo_creation_utils.cc | 385 HloInstruction* on_false, in MakeSelectHlo()
|
| D | tuple_points_to_analysis.cc | 454 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
|
| D | shape_inference.cc | 3264 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape()
3294 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferTupleSelectShape()
|
| D | hlo_evaluator.cc | 2067 const auto& on_false = GetEvaluatedLiteralFor(select->operand(2)); in HandleSelect() local
2085 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect() local
|
| D | layout_assignment_test.cc | 1535 const HloInstruction* on_false = FindInstruction(m.get(), "reshape.9717"); in TEST_F() local
|
| D | hlo_evaluator_test.cc | 296 auto on_false = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P() local
|
| /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/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 1233 Value on_false = transformed.on_false(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 378 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1236 XlaOp XlaBuilder::Select(XlaOp pred, XlaOp on_true, XlaOp on_false) { in Select()
3998 XlaOp Select(const XlaOp pred, const XlaOp on_true, const XlaOp on_false) { in Select()
|