| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | call_inliner_test.cc | 91 HloComputation* false_computation = local
96 HloInstruction::CreateCall(pred, {}, false_computation));
130 HloComputation* false_computation = local
135 HloInstruction::CreateCall(pred, {}, false_computation));
|
| H A D | call_graph_test.cc | 245 HloComputation* false_computation = local
258 false_computation));
273 UnorderedElementsAre(true_computation, false_computation));
282 const CallGraphNode& false_node = call_graph->GetNode(false_computation);
|
| H A D | layout_assignment_test.cc | 702 HloComputation* false_computation = local
705 result_tshape, pred, tuple, true_computation, tuple, false_computation));
713 const HloInstruction* false_root = false_computation->root_instruction();
|
| H A D | hlo_dataflow_analysis_test.cc | 1550 // false_computation(F32[] %false_param):
1558 // %constant2, false_computation)
1569 HloComputation* false_computation = local
1581 false_computation));
1620 // false_computation((F32[], F32[]) %false_param):
1631 // %tuple_operand, false_computation)
1654 HloComputation* false_computation = local
1668 false_computation));
|
| H A D | layout_assignment.cc | 598 HloComputation* false_computation = instruction->false_computation(); local
603 TF_RET_CHECK(false_computation->num_parameters() == 1);
607 FindOrDie(computation_layouts_, false_computation);
713 HloComputation* false_computation = instruction->false_computation(); local
726 false_computation->root_instruction()->shape()));
852 instruction->false_computation())));
|
| H A D | buffer_assignment_test.cc | 756 auto false_computation = module->AddEmbeddedComputation( local
767 r0f32_, pred, const1, true_computation, const2, false_computation));
775 GetInstructions(false_computation->root_instruction());
797 GetTopLevelAllocation(*buffers, false_computation->root_instruction());
|
| H A D | shape_inference.cc | 2220 const ProgramShape& false_computation) {
2243 if (false_computation.parameters_size() != 1) {
2244 return InvalidArgument("false_computation must take 1 argument; got %d.",
2245 false_computation.parameters_size());
2247 if (!ShapeUtil::Compatible(false_computation.parameters(0), false_operand)) {
2250 "false_operand: %s; false_computation: %s",
2252 ShapeUtil::HumanString(false_computation).c_str());
2256 "false_computation: got %s.",
2260 false_computation.result())) {
2263 "true_computation result: %s; false_computation resul
2217 InferConditionalShape( const Shape& predicate, const Shape& true_operand, const Shape& false_operand, const ProgramShape& true_computation, const ProgramShape& false_computation) argument
[all...] |
| H A D | user_computation.cc | 791 const UserComputation& false_computation) {
808 false_computation.version();
811 false_computation.ComputeProgramShape(false_computation_version));
2196 {conditional_request.false_computation(),
3296 HloComputation* false_computation = ResolveComputation( local
3297 conditional_request.false_computation(), false_computation_version);
3306 false_operand, false_computation));
788 AddConditionalInstruction( const ConditionalRequest& conditional_request, const UserComputation& true_computation, const UserComputation& false_computation) argument
|
| H A D | hlo_instruction.cc | 485 HloInstruction* false_computation_arg, HloComputation* false_computation) {
495 instruction->called_computations_.push_back(false_computation);
1412 false_computation());
1845 eq_computations(false_computation(), other.false_computation());
2064 HloComputation* HloInstruction::false_computation() const { function in class:xla::HloInstruction
2077 void HloInstruction::set_false_computation(HloComputation* false_computation) { argument
2082 called_computations_[kFalseComputationIndex] = false_computation;
2239 extra.push_back(StrCat("false_computation=",
2240 PrintName(false_computation()
482 CreateConditional( const Shape& shape, HloInstruction* pred, HloInstruction* true_computation_arg, HloComputation* true_computation, HloInstruction* false_computation_arg, HloComputation* false_computation) argument
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | conditional_test.cc | 177 auto false_computation = CreateR0ConstantComputation(12.0f); local
179 false_computation);
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | local_computation_builder.cc | 512 const LocalComputation& false_computation) {
515 false_computation.computation());
507 Conditional( const ComputationDataHandle& predicate, const ComputationDataHandle& true_operand, const LocalComputation& true_computation, const ComputationDataHandle& false_operand, const LocalComputation& false_computation) argument
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| H A D | hlo_parser.cc | 971 optional<HloComputation*> false_computation; local
974 attrs["false_computation"] = {/*required=*/true, AttrTy::kHloComputation,
975 &false_computation};
983 /*false_computation_arg=*/operands[2], *false_computation));
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | computation_builder.cc | 1258 const Computation& false_computation) {
1265 *request->mutable_false_computation() = false_computation.handle();
1253 Conditional( const ComputationDataHandle& predicate, const ComputationDataHandle& true_operand, const Computation& true_computation, const ComputationDataHandle& false_operand, const Computation& false_computation) argument
|