Home
last modified time | relevance | path

Searched defs:on_true (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dselect_test.cc38 auto on_true = ConstantR0<float>(&builder, 123.0f); in TEST_F() local
48 auto on_true = ConstantR0<int32_t>(&builder, -42); in TEST_F() local
58 auto on_true = ConstantR0<float>(&builder, 123.0f); in TEST_F() local
68 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
78 auto on_true = in TEST_F() local
95 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local
109 auto on_true = in TEST_F() local
125 auto on_true = in TEST_F() local
207 auto on_true = ConstantR1<float>(&builder, {11.0f, 22.0f, 33.0f, 44.0f}); in TEST_F() local
224 auto on_true = ConstantR1<float>(&builder, {11.0f, 22.0f, 33.0f, 44.0f}); in TEST_F() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconditional_code_motion_test.cc284 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
352 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
491 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
713 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
800 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
849 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
926 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
965 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
1006 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
1040 const HloComputation* on_true = conditional->branch_computation(0); in TEST_F() local
[all …]
H A Dhlo_creation_utils.cc488 HloInstruction* on_true, in MakeSelectHlo()
H A Dlayout_assignment_test.cc1538 const HloInstruction* on_true = FindInstruction(m.get(), "reshape.8493"); in TEST_F() local
H A Dshape_inference.cc3268 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape()
H A Dhlo_evaluator.cc3209 const auto& on_true = GetEvaluatedLiteralFor(select->operand(1)); in HandleSelect() local
H A Dhlo_evaluator_test.cc320 auto on_true = LiteralUtil::CreateR2<float>({{2.f, 4.f}, {4.f, 4.f}}); in TEST_P() local
H A Dalgebraic_simplifier.cc5953 auto on_true = select->mutable_operand(1); in HandleSelect() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc1292 XlaOp XlaBuilder::Select(XlaOp pred, XlaOp on_true, XlaOp on_false) { in Select()
4138 XlaOp Select(const XlaOp pred, const XlaOp on_true, const XlaOp on_false) { in Select()