| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner_test.cc | 91 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()
|
| D | layout_assignment.cc | 598 HloComputation* false_computation = instruction->false_computation(); in AddMandatoryConstraints() local
603 TF_RET_CHECK(false_computation->num_parameters() == 1); in AddMandatoryConstraints()
607 FindOrDie(computation_layouts_, false_computation); in AddMandatoryConstraints()
713 HloComputation* false_computation = instruction->false_computation(); in CheckConditionalLayout() local
726 false_computation->root_instruction()->shape())); in CheckConditionalLayout()
852 instruction->false_computation()))); in CheckLayouts()
|
| D | call_graph_test.cc | 245 HloComputation* false_computation = in TEST_F() local
258 false_computation)); in TEST_F()
273 UnorderedElementsAre(true_computation, false_computation)); in TEST_F()
282 const CallGraphNode& false_node = call_graph->GetNode(false_computation); in TEST_F()
|
| D | flatten_call_graph.cc | 55 CHECK_EQ(computation, instruction->false_computation()); in ReplaceCalledComputation()
|
| D | hlo_dataflow_analysis.cc | 370 conditional->false_computation()->root_instruction())}; in UpdateConditionalValueSet()
476 CHECK_EQ(parameter->parent(), conditional->false_computation()); in UpdateParameterValueSet()
639 add_to_worklist(user->false_computation()->parameter_instruction(0)); in Propagate()
|
| D | layout_assignment_test.cc | 702 HloComputation* false_computation = in TEST_F() local
705 result_tshape, pred, tuple, true_computation, tuple, false_computation)); in TEST_F()
713 const HloInstruction* false_root = false_computation->root_instruction(); in TEST_F()
|
| D | shape_inference.cc | 2220 const ProgramShape& false_computation) { in InferConditionalShape() argument
2243 if (false_computation.parameters_size() != 1) { in InferConditionalShape()
2245 false_computation.parameters_size()); in InferConditionalShape()
2247 if (!ShapeUtil::Compatible(false_computation.parameters(0), false_operand)) { in InferConditionalShape()
2252 ShapeUtil::HumanString(false_computation).c_str()); in InferConditionalShape()
2260 false_computation.result())) { in InferConditionalShape()
2265 ShapeUtil::HumanString(false_computation.result()).c_str()); in InferConditionalShape()
|
| D | hlo_instruction.h | 452 HloInstruction* false_computation_arg, HloComputation* false_computation);
776 HloComputation* false_computation() const;
778 void set_false_computation(HloComputation* false_computation);
|
| D | user_computation.h | 236 const UserComputation& false_computation);
|
| D | hlo_instruction.cc | 485 HloInstruction* false_computation_arg, HloComputation* false_computation) { in CreateConditional() argument
495 instruction->called_computations_.push_back(false_computation); in CreateConditional()
1412 false_computation()); in CloneWithNewOperands()
1845 eq_computations(false_computation(), other.false_computation()); in IdenticalSlowPath()
2064 HloComputation* HloInstruction::false_computation() const { in false_computation() function in xla::HloInstruction
2077 void HloInstruction::set_false_computation(HloComputation* false_computation) { in set_false_computation() argument
2082 called_computations_[kFalseComputationIndex] = false_computation; in set_false_computation()
2240 PrintName(false_computation()->name(), options))); in ExtraAttributesToString()
|
| D | shape_inference.h | 197 const ProgramShape& false_computation);
|
| D | hlo_ordering.cc | 198 conditional->false_computation())) { in UseIsBeforeValueDefinition()
|
| D | user_computation.cc | 791 const UserComputation& false_computation) { in AddConditionalInstruction() argument
808 false_computation.version(); in AddConditionalInstruction()
811 false_computation.ComputeProgramShape(false_computation_version)); in AddConditionalInstruction()
2196 {conditional_request.false_computation(), in GetEmbeddedComputations()
3296 HloComputation* false_computation = ResolveComputation( in Visit() local
3297 conditional_request.false_computation(), false_computation_version); in Visit()
3306 false_operand, false_computation)); in Visit()
|
| D | hlo_dataflow_analysis_test.cc | 1569 HloComputation* false_computation = in TEST_P() local
1581 false_computation)); in TEST_P()
1654 HloComputation* false_computation = in TEST_P() local
1668 false_computation)); in TEST_P()
|
| D | buffer_assignment_test.cc | 756 auto false_computation = module->AddEmbeddedComputation( in TEST_F() local
767 r0f32_, pred, const1, true_computation, const2, false_computation)); in TEST_F()
775 GetInstructions(false_computation->root_instruction()); in TEST_F()
797 GetTopLevelAllocation(*buffers, false_computation->root_instruction()); in TEST_F()
|
| D | service.cc | 1404 TF_ASSIGN_OR_RETURN(UserComputation * false_computation, in Op()
1406 arg->conditional_request().false_computation())); in Op()
1408 arg->conditional_request(), *true_computation, *false_computation); in Op()
|
| D | hlo_cost_analysis.cc | 523 ProcessSubcomputation(conditional->false_computation())); in HandleConditional()
|
| D | hlo_module.cc | 157 replacements, instruction->false_computation(), nullptr); in ReplaceComputations()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 177 auto false_computation = CreateR0ConstantComputation(12.0f); in XLA_TEST_F() local
179 false_computation); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 512 const LocalComputation& false_computation) { in Conditional() argument
515 false_computation.computation()); in Conditional()
|
| D | xla_client_test.py | 1244 false_computation = self._CreateConstantF32Computation()
1246 false_computation)
1255 false_computation = self._CreateConstantF32Computation()
1257 false_computation)
|
| D | xla_client.py | 970 false_computation): argument
987 false_computation.c_local_computation))
|
| D | local_computation_builder.h | 265 const LocalComputation& false_computation);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2362 conditional, conditional->false_computation()->root_instruction(), in CheckConditionalBuffersShareAllocation()
2379 conditional->false_computation()->parameter_instruction(0), index, in CheckConditionalBuffersShareAllocation()
2439 HloComputation* false_computation = hlo->false_computation(); in BuildConditionalThunk() local
2440 IrEmitterUnnested ir_emitter_false(hlo_module_config_, false_computation, in BuildConditionalThunk()
2442 TF_CHECK_OK(false_computation->root_instruction()->Accept(&ir_emitter_false)); in BuildConditionalThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2484 HloComputation* false_computation = conditional->false_computation(); in HandleConditional() local
2493 false_computation->root_instruction()->shape())) in HandleConditional()
2497 << ShapeUtil::HumanString(false_computation->root_instruction()->shape()); in HandleConditional()
2502 FindOrDie(emitted_functions_, false_computation); in HandleConditional()
|