Home
last modified time | relevance | path

Searched refs:false_computation (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconditional_simplifier_test.cc69 HloComputation* false_computation; in MakeConditional() local
81 false_computation = in MakeConditional()
97 false_param, false_computation)); in MakeConditional()
167 auto* false_computation = conditional->false_computation(); in TEST_F() local
168 auto token = false_computation->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
169 false_computation->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
H A Dcall_inliner_test.cc89 HloComputation* false_computation = in TEST_F() local
96 HloInstruction::CreateCall(pred, {}, false_computation)); in TEST_F()
130 HloComputation* false_computation = in TEST_F() local
135 HloInstruction::CreateCall(pred, {}, false_computation)); in TEST_F()
H A Dconditional_to_select.cc36 conditional->false_computation()->HasSideEffect()) { in DoConditionalToSelect()
53 conditional->false_computation())); in DoConditionalToSelect()
H A Dcall_graph_test.cc257 HloComputation* false_computation = in TEST_F() local
270 false_computation)); in TEST_F()
286 UnorderedElementsAre(true_computation, false_computation)); in TEST_F()
296 const CallGraphNode& false_node = call_graph->GetNode(false_computation); in TEST_F()
H A Dlayout_assignment_test.cc758 HloComputation* false_computation = in TEST_F() local
761 result_tshape, pred, tuple, true_computation, tuple, false_computation)); in TEST_F()
769 const HloInstruction* false_root = false_computation->root_instruction(); in TEST_F()
H A Dbuffer_assignment_test.cc1110 auto false_computation = module->AddEmbeddedComputation( in TEST_F() local
1121 r0f32_, pred, const1, true_computation, const2, false_computation)); in TEST_F()
1129 GetInstructions(false_computation->root_instruction()); in TEST_F()
1151 GetTopLevelAllocation(*buffers, false_computation->root_instruction()); in TEST_F()
H A Dhlo_dataflow_analysis_test.cc1658 HloComputation* false_computation = in TEST_P() local
1670 false_computation)); in TEST_P()
1750 HloComputation* false_computation = in TEST_P() local
1764 false_computation)); in TEST_P()
H A Dhlo_instruction.h1051 HloInstruction* false_computation_arg, HloComputation* false_computation);
1475 HloComputation* false_computation() const;
H A Dhlo_instruction.cc1475 HloInstruction* false_computation_arg, HloComputation* false_computation) { in CreateConditional() argument
1485 instruction->called_computations_.push_back(false_computation); in CreateConditional()
2866 HloComputation* HloInstruction::false_computation() const { in false_computation() function in xla::HloInstruction
3220 PrintNameInternal(false_computation()->name(), options))); in ExtraAttributesToString()
3291 false_computation()->ToString(new_options))); in ExtraAttributesToString()
H A Ddynamic_dimension_inference_test.cc932 conditional_hlo->false_computation()->instructions()) { in TEST_F()
H A Dhlo_parser.cc2524 optional<HloComputation*> false_computation; in CreateInstruction() local
2539 /*required=*/true, AttrTy::kHloComputation, &false_computation}; in CreateInstruction()
2553 branch_computations.emplace({*true_computation, *false_computation}); in CreateInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dvalue_inference_test.cc359 auto false_computation = false_builder.Build().ValueOrDie(); in TEST_F() local
365 false_computation); in TEST_F()
418 auto false_computation = false_builder.Build().ValueOrDie(); in TEST_F() local
424 false_computation); in TEST_F()
451 auto false_computation = false_builder.Build().ValueOrDie(); in TEST_F() local
457 false_computation); in TEST_F()
485 auto false_computation = false_builder.Build().ValueOrDie(); in TEST_F() local
492 Conditional(pred, constant, true_computation, param, false_computation); in TEST_F()
547 auto false_computation = false_builder.Build().ValueOrDie(); in TEST_F() local
553 Conditional(pred, param, true_computation, param, false_computation); in TEST_F()
H A Doutfeed_in_nested_computation_test.cc135 TF_ASSERT_OK_AND_ASSIGN(XlaComputation false_computation, [&] { in XLA_TEST_F()
144 /*false_computation=*/false_computation); in XLA_TEST_F()
H A Dconditional_test.cc187 auto false_computation = CreateR0ConstantComputation(12.0f); in XLA_TEST_F() local
188 Conditional(pred, operands, true_computation, operands, false_computation); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
H A Dif.mlir19 …(pred[] %[[VAL1]], f32[] %[[A0]], f32[] %[[A0]]), true_computation=[[R0]], false_computation=[[R1]]
70 … f32[]) %[[TUPLE1]], (f32[], f32[]) %[[TUPLE2]]), true_computation=[[R0]], false_computation=[[R1]]
110 … f32[]) %[[TUPLE1]], (f32[], f32[]) %[[TUPLE2]]), true_computation=[[R0]], false_computation=[[R1]]
150 …32[], f32[], f32[]) %[[TUPLE1]], () %[[TUPLE2]]), true_computation=[[R0]], false_computation=[[R1]]
190 … %[[TUPLE1]], (f32[], f32[], f32[]) %[[TUPLE2]]), true_computation=[[R0]], false_computation=[[R1]]
226 …[] %[[PRED:.+]], () %[[TUPLE1]], () %[[TUPLE2]]), true_computation=[[R0]], false_computation=[[R1]]
281 …pred[] %{{.+}}, f32[] %[[CST2]], () %[[TUPLE2]]), true_computation=[[R0]], false_computation=[[R1]]
297 …0]], () %[[TUPLE1]], (f32[], f32[]) %[[TUPLE2]]), true_computation=[[R2]], false_computation=[[R3]]
H A Dif_conditional.hlotxt42 …al(%compare.4, %tuple.5, %tuple.5), true_computation=%then_branch0, false_computation=%else_branch…
109 …al(%compare.4, %tuple.5, %tuple.6), true_computation=%then_branch1, false_computation=%else_branch1
147 …ditional(%compare.4, %arg0, %arg0), true_computation=%then_branch2, false_computation=%else_branch2
168 ….14, f32[] %constant.15, () %tuple.16), true_computation=%region_1.7, false_computation=%region_2.9
204 …) %tuple.5, (f32[], f32[]) %tuple.6), true_computation=%region_0.12, false_computation=%region_3.19
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Ddynamic_shaped_ops.cc126 const XlaComputation& false_computation) { in DynamicConditional() argument
130 auto false_shape = false_computation.GetProgramShape().value().result(); in DynamicConditional()
134 false_operand, false_computation); in DynamicConditional()
160 true_shape, false_computation)); in DynamicConditional()
H A Ddynamic_shaped_ops.h35 const XlaComputation& false_computation);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_client_test.py2077 false_computation = self._CreateConstantComputation(
2080 false_computation)
2089 false_computation = self._CreateConstantComputation(
2092 false_computation)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A Dops.pyi149 false_computation: XlaComputation) -> XlaOp: ...
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h860 const XlaComputation& false_computation);
1459 const XlaComputation& false_computation);
2578 const XlaComputation& false_computation);
H A Dxla_builder.cc2499 const XlaComputation& false_computation) { in Conditional() argument
2511 return ConditionalImpl(predicate, {&true_computation, &false_computation}, in Conditional()
4880 const XlaComputation& false_computation) { in Conditional() argument
4883 false_computation); in Conditional()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/
H A Doperation_semantics.md692 false_computation)` </b>
702 `false_computation` | `XlaComputation` | XlaComputation of type \\(T_1 \to S\\)
704 Executes `true_computation` if `pred` is `true`, `false_computation` if `pred`
709 `false_computation` must take in a single argument of type \\(T_1\\) and will be
711 returned value of `true_computation` and `false_computation` must be the same.
715 Note that only one of `true_computation` and `false_computation` will be
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc385 conditional->false_computation()->Accept(&candidates_for_false)); in HandleConditional()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc2875 HloComputation* false_computation; in HandleSingleDevice() local
2880 false_computation = module_->AddEmbeddedComputation(false_b.Build(root)); in HandleSingleDevice()
2886 false_computation)); in HandleSingleDevice()

12