| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 597 HloComputation* true_computation = instruction->true_computation(); in AddMandatoryConstraints() local
602 TF_RET_CHECK(true_computation->num_parameters() == 1); in AddMandatoryConstraints()
605 FindOrDie(computation_layouts_, true_computation); in AddMandatoryConstraints()
712 HloComputation* true_computation = instruction->true_computation(); in CheckConditionalLayout() local
722 true_computation->root_instruction()->shape())); in CheckConditionalLayout()
850 FindOrDie(computation_layouts_, instruction->true_computation()), in CheckLayouts()
|
| D | call_graph_test.cc | 243 HloComputation* true_computation = in TEST_F() local
257 kScalarShape, pred, const1, true_computation, const2, in TEST_F()
273 UnorderedElementsAre(true_computation, false_computation)); in TEST_F()
277 const CallGraphNode& true_node = call_graph->GetNode(true_computation); in TEST_F()
|
| D | flatten_call_graph.cc | 52 if (computation == instruction->true_computation()) { in ReplaceCalledComputation()
|
| D | hlo_dataflow_analysis.cc | 368 conditional->true_computation()->root_instruction()), in UpdateConditionalValueSet()
473 if (parameter->parent() == conditional->true_computation()) { in UpdateParameterValueSet()
636 add_to_worklist(user->true_computation()->parameter_instruction(0)); in Propagate()
|
| D | shape_inference.cc | 2219 const Shape& false_operand, const ProgramShape& true_computation, in InferConditionalShape() argument
2226 if (true_computation.parameters_size() != 1) { in InferConditionalShape()
2228 true_computation.parameters_size()); in InferConditionalShape()
2230 if (!ShapeUtil::Compatible(true_computation.parameters(0), true_operand)) { in InferConditionalShape()
2235 ShapeUtil::HumanString(true_computation).c_str()); in InferConditionalShape()
2259 if (!ShapeUtil::Compatible(true_computation.result(), in InferConditionalShape()
2264 ShapeUtil::HumanString(true_computation.result()).c_str(), in InferConditionalShape()
2272 return true_computation.result(); in InferConditionalShape()
|
| D | layout_assignment_test.cc | 689 HloComputation* true_computation = in TEST_F() local
705 result_tshape, pred, tuple, true_computation, tuple, false_computation)); in TEST_F()
712 const HloInstruction* true_root = true_computation->root_instruction(); in TEST_F()
|
| D | hlo_instruction.h | 451 HloInstruction* true_computation_arg, HloComputation* true_computation,
775 HloComputation* true_computation() const;
777 void set_true_computation(HloComputation* true_computation);
|
| D | user_computation.h | 235 const UserComputation& true_computation,
|
| D | hlo_instruction.cc | 484 HloInstruction* true_computation_arg, HloComputation* true_computation, in CreateConditional() argument
494 instruction->called_computations_.push_back(true_computation); in CreateConditional()
1411 true_computation(), new_operands[2], in CloneWithNewOperands()
1844 return eq_computations(true_computation(), other.true_computation()) && in IdenticalSlowPath()
2059 HloComputation* HloInstruction::true_computation() const { in true_computation() function in xla::HloInstruction
2069 void HloInstruction::set_true_computation(HloComputation* true_computation) { in set_true_computation() argument
2074 called_computations_[kTrueComputationIndex] = true_computation; in set_true_computation()
2238 PrintName(true_computation()->name(), options))); in ExtraAttributesToString()
|
| D | shape_inference.h | 196 const Shape& false_operand, const ProgramShape& true_computation,
|
| D | hlo_ordering.cc | 192 conditional->true_computation())) { in UseIsBeforeValueDefinition()
|
| D | user_computation.cc | 790 const UserComputation& true_computation, in AddConditionalInstruction() argument
802 true_computation.version(); in AddConditionalInstruction()
805 true_computation.ComputeProgramShape(true_computation_version)); in AddConditionalInstruction()
2192 conditional_request.true_computation(), in GetEmbeddedComputations()
3292 HloComputation* true_computation = ResolveComputation( in Visit() local
3293 conditional_request.true_computation(), true_computation_version); in Visit()
3305 request.output_shape(), predicate, true_operand, true_computation, in Visit()
|
| D | hlo_dataflow_analysis_test.cc | 1563 HloComputation* true_computation = in TEST_P() local
1580 scalar_shape_, pred, constant1, true_computation, constant2, in TEST_P()
1642 HloComputation* true_computation = in TEST_P() local
1667 scalar_shape_, pred, tuple_operand, true_computation, tuple_operand, in TEST_P()
|
| D | buffer_assignment_test.cc | 754 auto true_computation = module->AddEmbeddedComputation( in TEST_F() local
767 r0f32_, pred, const1, true_computation, const2, false_computation)); in TEST_F()
773 GetInstructions(true_computation->root_instruction()); in TEST_F()
795 GetTopLevelAllocation(*buffers, true_computation->root_instruction()); in TEST_F()
|
| D | service.cc | 1401 TF_ASSIGN_OR_RETURN(UserComputation * true_computation, in Op()
1403 arg->conditional_request().true_computation())); in Op()
1408 arg->conditional_request(), *true_computation, *false_computation); in Op()
|
| D | hlo_cost_analysis.cc | 521 ProcessSubcomputation(conditional->true_computation())); in HandleConditional()
|
| D | hlo_module.cc | 151 replacements, instruction->true_computation(), nullptr); in ReplaceComputations()
|
| D | buffer_assignment.cc | 1342 conditional_hlo->true_computation()->root_instruction(), in BuildColocatedBufferSets()
1364 conditional_hlo->true_computation()->parameter_instruction(0), in BuildColocatedBufferSets()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conditional_test.cc | 176 auto true_computation = CreateR0ConstantComputation(56.0f); in XLA_TEST_F() local
178 auto result = builder.Conditional(pred, operands, true_computation, operands, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 510 const LocalComputation& true_computation, in Conditional() argument
514 true_computation.computation(), false_operand, in Conditional()
|
| D | xla_client_test.py | 1242 true_computation = self._CreateMulF32By2Computation()
1245 c.Conditional(pred, true_operand, true_computation, false_operand,
1253 true_computation = self._CreateMulF32By2Computation()
1256 c.Conditional(pred, true_operand, true_computation, false_operand,
|
| D | xla_client.py | 969 def Conditional(self, pred, true_operand, true_computation, false_operand, argument
985 true_computation.c_local_computation,
|
| D | local_computation_builder.h | 263 const LocalComputation& true_computation,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2359 conditional, conditional->true_computation()->root_instruction(), in CheckConditionalBuffersShareAllocation()
2371 conditional->true_computation()->parameter_instruction(0), index, in CheckConditionalBuffersShareAllocation()
2434 HloComputation* true_computation = hlo->true_computation(); in BuildConditionalThunk() local
2435 IrEmitterUnnested ir_emitter_true(hlo_module_config_, true_computation, in BuildConditionalThunk()
2437 TF_CHECK_OK(true_computation->root_instruction()->Accept(&ir_emitter_true)); in BuildConditionalThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2483 HloComputation* true_computation = conditional->true_computation(); in HandleConditional() local
2486 true_computation->root_instruction()->shape())) in HandleConditional()
2490 << ShapeUtil::HumanString(true_computation->root_instruction()->shape()); in HandleConditional()
2500 FindOrDie(emitted_functions_, true_computation); in HandleConditional()
|