/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | conditional_simplifier_test.cc | 69 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 D | call_inliner_test.cc | 89 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 D | conditional_to_select.cc | 36 conditional->false_computation()->HasSideEffect()) { in DoConditionalToSelect() 53 conditional->false_computation())); in DoConditionalToSelect()
|
H A D | call_graph_test.cc | 257 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 D | layout_assignment_test.cc | 758 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 D | buffer_assignment_test.cc | 1110 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 D | hlo_dataflow_analysis_test.cc | 1658 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 D | hlo_instruction.h | 1051 HloInstruction* false_computation_arg, HloComputation* false_computation); 1475 HloComputation* false_computation() const;
|
H A D | hlo_instruction.cc | 1475 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 D | dynamic_dimension_inference_test.cc | 932 conditional_hlo->false_computation()->instructions()) { in TEST_F()
|
H A D | hlo_parser.cc | 2524 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 D | value_inference_test.cc | 359 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 D | outfeed_in_nested_computation_test.cc | 135 TF_ASSERT_OK_AND_ASSIGN(XlaComputation false_computation, [&] { in XLA_TEST_F() 144 /*false_computation=*/false_computation); in XLA_TEST_F()
|
H A D | conditional_test.cc | 187 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 D | if.mlir | 19 …(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 D | if_conditional.hlotxt | 42 …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 D | dynamic_shaped_ops.cc | 126 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 D | dynamic_shaped_ops.h | 35 const XlaComputation& false_computation);
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
H A D | xla_client_test.py | 2077 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 D | ops.pyi | 149 false_computation: XlaComputation) -> XlaOp: ...
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | xla_builder.h | 860 const XlaComputation& false_computation); 1459 const XlaComputation& false_computation); 2578 const XlaComputation& false_computation);
|
H A D | xla_builder.cc | 2499 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 D | operation_semantics.md | 692 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 D | cpu_compiler.cc | 385 conditional->false_computation()->Accept(&candidates_for_false)); in HandleConditional()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
H A D | spmd_partitioner.cc | 2875 HloComputation* false_computation; in HandleSingleDevice() local 2880 false_computation = module_->AddEmbeddedComputation(false_b.Build(root)); in HandleSingleDevice() 2886 false_computation)); in HandleSingleDevice()
|