Home
last modified time | relevance | path

Searched refs:true_computation (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_simplifier_test.cc51 HloComputation* true_computation; in MakeConditional() local
64 true_computation = in MakeConditional()
96 ShapeUtil::MakeShape(S32, {}), false_instrn, one, true_computation, in MakeConditional()
138 auto* true_computation = conditional->true_computation(); in TEST_F() local
139 auto* token = true_computation->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
140 auto* send = true_computation->AddInstruction(HloInstruction::CreateSend( in TEST_F()
141 true_computation->AddInstruction( in TEST_F()
144 true_computation->AddInstruction(HloInstruction::CreateSendDone(send)); in TEST_F()
154 auto* true_computation = conditional->true_computation(); in TEST_F() local
155 auto* token = true_computation->AddInstruction(HloInstruction::CreateToken()); in TEST_F()
[all …]
Dconditional_to_select.cc35 if (conditional->true_computation()->HasSideEffect() || in DoConditionalToSelect()
48 conditional->true_computation())); in DoConditionalToSelect()
Dcall_graph_test.cc255 HloComputation* true_computation = in TEST_F() local
269 kScalarShape, pred, const1, true_computation, const2, in TEST_F()
286 UnorderedElementsAre(true_computation, false_computation)); in TEST_F()
290 const CallGraphNode& true_node = call_graph->GetNode(true_computation); in TEST_F()
Dlayout_assignment_test.cc742 HloComputation* true_computation = in TEST_F() local
761 result_tshape, pred, tuple, true_computation, tuple, false_computation)); in TEST_F()
768 const HloInstruction* true_root = true_computation->root_instruction(); in TEST_F()
Dbuffer_assignment_test.cc1108 auto true_computation = module->AddEmbeddedComputation( in TEST_F() local
1121 r0f32_, pred, const1, true_computation, const2, false_computation)); in TEST_F()
1127 GetInstructions(true_computation->root_instruction()); in TEST_F()
1149 GetTopLevelAllocation(*buffers, true_computation->root_instruction()); in TEST_F()
Dhlo_instruction.h1050 HloInstruction* true_computation_arg, HloComputation* true_computation,
1474 HloComputation* true_computation() const;
Dhlo_dataflow_analysis_test.cc1652 HloComputation* true_computation = in TEST_P() local
1669 scalar_shape_, pred, constant1, true_computation, constant2, in TEST_P()
1738 HloComputation* true_computation = in TEST_P() local
1763 scalar_shape_, pred, tuple_operand, true_computation, tuple_operand, in TEST_P()
Dhlo_instruction.cc1474 HloInstruction* true_computation_arg, HloComputation* true_computation, in CreateConditional() argument
1484 instruction->called_computations_.push_back(true_computation); in CreateConditional()
2860 HloComputation* HloInstruction::true_computation() const { in true_computation() function in xla::HloInstruction
3217 PrintNameInternal(true_computation()->name(), options))); in ExtraAttributesToString()
3289 true_computation()->ToString(new_options))); in ExtraAttributesToString()
Ddynamic_dimension_inference_test.cc922 conditional_hlo->true_computation()->instructions()) { in TEST_F()
Dhlo_parser.cc2523 optional<HloComputation*> true_computation; in CreateInstruction() local
2537 &true_computation}; in CreateInstruction()
2553 branch_computations.emplace({*true_computation, *false_computation}); in CreateInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dvalue_inference_test.cc354 auto true_computation = true_builder.Build().ValueOrDie(); in TEST_F() local
364 auto cond = Conditional(parameter, constant, true_computation, constant, in TEST_F()
413 auto true_computation = true_builder.Build().ValueOrDie(); in TEST_F() local
423 auto cond = Conditional(parameter, constant, true_computation, constant, in TEST_F()
446 auto true_computation = true_builder.Build().ValueOrDie(); in TEST_F() local
456 auto cond = Conditional(pred, constant, true_computation, constant, in TEST_F()
480 auto true_computation = true_builder.Build().ValueOrDie(); in TEST_F() local
492 Conditional(pred, constant, true_computation, param, false_computation); in TEST_F()
539 auto true_computation = true_builder.Build().ValueOrDie(); in TEST_F() local
553 Conditional(pred, param, true_computation, param, false_computation); in TEST_F()
Doutfeed_in_nested_computation_test.cc127 TF_ASSERT_OK_AND_ASSIGN(XlaComputation true_computation, [&] { in XLA_TEST_F()
143 /*true_computation=*/true_computation, /*false_operand=*/pred, in XLA_TEST_F()
Dconditional_test.cc186 auto true_computation = CreateR0ConstantComputation(56.0f); in XLA_TEST_F() local
188 Conditional(pred, operands, true_computation, operands, false_computation); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dif.mlir19 … f32[] conditional(pred[] %[[VAL1]], f32[] %[[A0]], f32[] %[[A0]]), true_computation=[[R0]], false…
70 …PRED:.+]], (f32[], f32[]) %[[TUPLE1]], (f32[], f32[]) %[[TUPLE2]]), true_computation=[[R0]], false…
110 …]], (f32[], f32[], f32[]) %[[TUPLE1]], (f32[], f32[]) %[[TUPLE2]]), true_computation=[[R0]], false…
150 …] %[[PRED:.+]], (f32[], f32[], f32[]) %[[TUPLE1]], () %[[TUPLE2]]), true_computation=[[R0]], false…
190 …] %[[PRED:.+]], () %[[TUPLE1]], (f32[], f32[], f32[]) %[[TUPLE2]]), true_computation=[[R0]], false…
226 …) conditional(pred[] %[[PRED:.+]], () %[[TUPLE1]], () %[[TUPLE2]]), true_computation=[[R0]], false…
281 …f32[] conditional(pred[] %{{.+}}, f32[] %[[CST2]], () %[[TUPLE2]]), true_computation=[[R0]], false…
297 …tional(pred[] %[[A0]], () %[[TUPLE1]], (f32[], f32[]) %[[TUPLE2]]), true_computation=[[R2]], false…
Dif_conditional.hlotxt42 …%conditional.16 = (f32[]) conditional(%compare.4, %tuple.5, %tuple.5), true_computation=%then_bran…
109 …((f32[],f32[]), f32[]) conditional(%compare.4, %tuple.5, %tuple.6), true_computation=%then_branch1…
147 …ROOT %conditional.16 = f32[] conditional(%compare.4, %arg0, %arg0), true_computation=%then_branch2…
168 …conditional(pred[] %constant.14, f32[] %constant.15, () %tuple.16), true_computation=%region_1.7, …
204 …conditional(pred[] %Arg_0.1, () %tuple.5, (f32[], f32[]) %tuple.6), true_computation=%region_0.12,…
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Ddynamic_shaped_ops.cc124 const XlaComputation& true_computation, in DynamicConditional() argument
128 auto true_shape = true_computation.GetProgramShape().value().result(); in DynamicConditional()
133 return xla::Conditional(predicate, true_operand, true_computation, in DynamicConditional()
154 false_shape, true_computation)); in DynamicConditional()
Ddynamic_shaped_ops.h33 const XlaComputation& true_computation,
/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
Dops.pyi147 true_computation: XlaComputation,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py2075 true_computation = self._CreateMulBy2Computation(np.float32)
2079 ops.Conditional(pred, true_operand, true_computation, false_operand,
2087 true_computation = self._CreateMulBy2Computation(np.float32)
2091 ops.Conditional(pred, true_operand, true_computation, false_operand,
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc908 {instruction->true_computation()->root_instruction()}, &rets)); in ImportInstructionImpl()
917 TF_RETURN_IF_ERROR(ImportAsRegion(*instruction->true_computation(), in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h859 const XlaComputation& true_computation, XlaOp false_operand,
1457 const XlaComputation& true_computation,
2577 const XlaComputation& true_computation, XlaOp false_operand,
Dxla_builder.cc2497 const XlaComputation& true_computation, in Conditional() argument
2511 return ConditionalImpl(predicate, {&true_computation, &false_computation}, in Conditional()
4878 const XlaComputation& true_computation, in Conditional() argument
4882 true_computation, false_operand, in Conditional()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md691 <b> `Conditional(pred, true_operand, true_computation, false_operand,
700 `true_computation` | `XlaComputation` | XlaComputation of type \\(T_0 \to S\\)
704 Executes `true_computation` if `pred` is `true`, `false_computation` if `pred`
707 The `true_computation` must take in a single argument of type \\(T_0\\) and will
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
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc380 conditional->true_computation()->Accept(&candidates_for_true)); in HandleConditional()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2860 HloComputation* true_computation; in HandleSingleDevice() local
2871 true_computation = module_->AddEmbeddedComputation(true_b.Build(root)); in HandleSingleDevice()
2885 hlo->shape(), pred, operand, true_computation, operand, in HandleSingleDevice()

12