| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier_test.cc | 68 HloComputation* false_computation; in MakeConditional() local
80 false_computation = in MakeConditional()
93 false_param, false_computation)); in MakeConditional()
153 auto* false_computation = conditional->false_computation(); in TEST_F() local
154 auto token = false_computation->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
155 false_computation->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
|
| D | call_inliner_test.cc | 90 HloComputation* false_computation = in TEST_F() local
97 HloInstruction::CreateCall(pred, {}, false_computation)); in TEST_F()
131 HloComputation* false_computation = in TEST_F() local
136 HloInstruction::CreateCall(pred, {}, false_computation)); in TEST_F()
|
| D | call_graph_test.cc | 252 HloComputation* false_computation = in TEST_F() local
265 false_computation)); in TEST_F()
281 UnorderedElementsAre(true_computation, false_computation)); in TEST_F()
291 const CallGraphNode& false_node = call_graph->GetNode(false_computation); in TEST_F()
|
| D | layout_assignment_test.cc | 798 HloComputation* false_computation = in TEST_F() local
801 result_tshape, pred, tuple, true_computation, tuple, false_computation)); in TEST_F()
809 const HloInstruction* false_root = false_computation->root_instruction(); in TEST_F()
|
| D | hlo_instruction.h | 718 HloInstruction* false_computation_arg, HloComputation* false_computation);
1075 HloComputation* false_computation() const;
|
| D | buffer_assignment_test.cc | 919 auto false_computation = module->AddEmbeddedComputation( in TEST_F() local
930 r0f32_, pred, const1, true_computation, const2, false_computation)); in TEST_F()
938 GetInstructions(false_computation->root_instruction()); in TEST_F()
960 GetTopLevelAllocation(*buffers, false_computation->root_instruction()); in TEST_F()
|
| D | hlo_instruction.cc | 955 HloInstruction* false_computation_arg, HloComputation* false_computation) { in CreateConditional() argument
965 instruction->called_computations_.push_back(false_computation); in CreateConditional()
2063 HloComputation* HloInstruction::false_computation() const { in false_computation() function in xla::HloInstruction
2296 PrintName(false_computation()->name(), options))); in ExtraAttributesToString()
2341 false_computation()->ToString(new_options))); in ExtraAttributesToString()
|
| D | hlo_dataflow_analysis_test.cc | 1603 HloComputation* false_computation = in TEST_P() local
1615 false_computation)); in TEST_P()
1695 HloComputation* false_computation = in TEST_P() local
1709 false_computation)); in TEST_P()
|
| D | hlo_parser.cc | 1450 optional<HloComputation*> false_computation; in ParseInstructionRhs() local
1461 /*required=*/true, AttrTy::kHloComputation, &false_computation}; in ParseInstructionRhs()
1471 branch_computations.emplace({*true_computation, *false_computation}); in ParseInstructionRhs()
|
| D | hlo_parser_test.cc | 1273 … conditional(constant, constant.1, constant.2), true_computation=Negate, false_computation=Identity in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| 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()
|
| D | conditional_test.cc | 185 auto false_computation = CreateR0ConstantComputation(12.0f); in XLA_TEST_F() local
186 Conditional(pred, operands, true_computation, operands, false_computation); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1633 false_computation = self._CreateConstantF32Computation()
1635 false_computation)
1644 false_computation = self._CreateConstantF32Computation()
1646 false_computation)
|
| D | local_computation_builder.cc | 698 const Computation& false_computation) { in Conditional() argument
701 false_computation.computation()); in Conditional()
|
| D | local_computation_builder.h | 350 const Computation& false_computation);
|
| D | xla_client.py | 1558 false_computation): argument
1571 false_computation.computation)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 534 const XlaComputation& false_computation);
961 const XlaComputation& false_computation);
1799 const XlaComputation& false_computation);
|
| D | xla_builder.cc | 1894 const XlaComputation& false_computation) { in Conditional() argument
1897 return Conditional(predicate, {&true_computation, &false_computation}, in Conditional()
3405 const XlaComputation& false_computation) { in Conditional() argument
3408 false_computation); in Conditional()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 541 false_computation)` </b>
549 `false_computation` | `XlaComputation` | XlaComputation of type $$ T_1 \to S $$
551 Executes `true_computation` if `pred` is `true`, `false_computation` if `pred`
556 `false_computation` must take in a single argument of type $$ T_1 $$ and will be
558 returned value of `true_computation` and `false_computation` must be the same.
560 Note that only one of `true_computation` and `false_computation` will be
|