| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier_test.cc | 50 HloComputation* true_computation; in MakeConditional() local
63 true_computation = in MakeConditional()
92 ShapeUtil::MakeShape(S32, {}), false_instrn, one, true_computation, in MakeConditional()
124 auto* true_computation = conditional->true_computation(); in TEST_F() local
125 auto* token = true_computation->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
126 auto* send = true_computation->AddInstruction(HloInstruction::CreateSend( in TEST_F()
127 true_computation->AddInstruction( in TEST_F()
130 true_computation->AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
140 auto* true_computation = conditional->true_computation(); in TEST_F() local
141 auto* token = true_computation->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
[all …]
|
| D | call_graph_test.cc | 250 HloComputation* true_computation = in TEST_F() local
264 kScalarShape, pred, const1, true_computation, const2, in TEST_F()
281 UnorderedElementsAre(true_computation, false_computation)); in TEST_F()
285 const CallGraphNode& true_node = call_graph->GetNode(true_computation); in TEST_F()
|
| D | layout_assignment_test.cc | 782 HloComputation* true_computation = in TEST_F() local
801 result_tshape, pred, tuple, true_computation, tuple, false_computation)); in TEST_F()
808 const HloInstruction* true_root = true_computation->root_instruction(); in TEST_F()
|
| D | hlo_instruction.h | 717 HloInstruction* true_computation_arg, HloComputation* true_computation,
1074 HloComputation* true_computation() const;
|
| D | buffer_assignment_test.cc | 917 auto true_computation = module->AddEmbeddedComputation( in TEST_F() local
930 r0f32_, pred, const1, true_computation, const2, false_computation)); in TEST_F()
936 GetInstructions(true_computation->root_instruction()); in TEST_F()
958 GetTopLevelAllocation(*buffers, true_computation->root_instruction()); in TEST_F()
|
| D | hlo_instruction.cc | 954 HloInstruction* true_computation_arg, HloComputation* true_computation, in CreateConditional() argument
964 instruction->called_computations_.push_back(true_computation); in CreateConditional()
2057 HloComputation* HloInstruction::true_computation() const { in true_computation() function in xla::HloInstruction
2293 PrintName(true_computation()->name(), options))); in ExtraAttributesToString()
2339 true_computation()->ToString(new_options))); in ExtraAttributesToString()
|
| D | hlo_dataflow_analysis_test.cc | 1597 HloComputation* true_computation = in TEST_P() local
1614 scalar_shape_, pred, constant1, true_computation, constant2, in TEST_P()
1683 HloComputation* true_computation = in TEST_P() local
1708 scalar_shape_, pred, tuple_operand, true_computation, tuple_operand, in TEST_P()
|
| D | hlo_parser.cc | 1449 optional<HloComputation*> true_computation; in ParseInstructionRhs() local
1459 &true_computation}; in ParseInstructionRhs()
1471 branch_computations.emplace({*true_computation, *false_computation}); in ParseInstructionRhs()
|
| D | hlo_parser_test.cc | 1273 …ROOT conditional = f32[] conditional(constant, constant.1, constant.2), true_computation=Negate, f… in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 127 TF_ASSERT_OK_AND_ASSIGN(XlaComputation true_computation, [&] { in XLA_TEST_F()
143 /*true_computation=*/true_computation, /*false_operand=*/pred, in XLA_TEST_F()
|
| D | conditional_test.cc | 184 auto true_computation = CreateR0ConstantComputation(56.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 | 1631 true_computation = self._CreateMulF32By2Computation()
1634 c.Conditional(pred, true_operand, true_computation, false_operand,
1642 true_computation = self._CreateMulF32By2Computation()
1645 c.Conditional(pred, true_operand, true_computation, false_operand,
|
| D | local_computation_builder.cc | 696 const Computation& true_computation, in Conditional() argument
700 true_computation.computation(), false_operand.op(), in Conditional()
|
| D | local_computation_builder.h | 348 const Computation& true_computation,
|
| D | xla_client.py | 1557 def Conditional(self, pred, true_operand, true_computation, false_operand, argument
1570 true_computation.computation, false_operand,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 532 const XlaComputation& true_computation,
959 const XlaComputation& true_computation,
1797 const XlaComputation& true_computation,
|
| D | xla_builder.cc | 1892 const XlaComputation& true_computation, in Conditional() argument
1897 return Conditional(predicate, {&true_computation, &false_computation}, in Conditional()
3403 const XlaComputation& true_computation, in Conditional() argument
3407 true_computation, false_operand, in Conditional()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 540 <b> `Conditional(pred, true_operand, true_computation, false_operand,
547 `true_computation` | `XlaComputation` | XlaComputation of type $$ T_0 \to S$$
551 Executes `true_computation` if `pred` is `true`, `false_computation` if `pred`
554 The `true_computation` must take in a single argument of type $$ T_0 $$ and will
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
|