Home
last modified time | relevance | path

Searched refs:fused_parameters (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc514 auto& fused_parameters = fusion->fused_parameters(); in HandleFusion() local
515 if (fused_parameters.size() != fusion->operand_count()) { in HandleFusion()
519 fused_parameters.size(), fusion->operand_count(), in HandleFusion()
522 for (HloInstruction* fused_param : fused_parameters) { in HandleFusion()
1193 const std::vector<HloInstruction*>& fused_parameters = in CheckFusionInstruction() local
1194 fusion->fused_parameters(); in CheckFusionInstruction()
1196 std::vector<bool> parameter_owned(fused_parameters.size(), false); in CheckFusionInstruction()
1205 for (int i = 0; i < fused_parameters.size(); ++i) { in CheckFusionInstruction()
1206 if (fused_parameters[i] == instruction) { in CheckFusionInstruction()
1253 CHECK_EQ(fusion->operands().size(), fused_parameters.size()); in CheckFusionInstruction()
[all …]
Dtuple_points_to_analysis.cc605 user->fused_parameters(), [&](HloInstruction* fused_param) { in DoesNotUseOperandBuffer()
608 CHECK(it != user->fused_parameters().end()); in DoesNotUseOperandBuffer()
670 const auto& fused_params = fusion->fused_parameters(); in HasUniqueFusedUseOfOperandAt()
Dhlo_instructions.cc1346 const std::vector<HloInstruction*>& HloFusionInstruction::fused_parameters() in fused_parameters() function in xla::HloFusionInstruction
1413 const std::vector<HloInstruction*>& fused_parameters = in CloneAndFuseInternal() local
1418 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal()
1442 const std::vector<HloInstruction*>& fused_parameters = in CloneAndFuseInternal() local
1453 CHECK_EQ(operands().size(), fused_parameters.size()); in CloneAndFuseInternal()
1457 fused_param = fused_parameters[i]; in CloneAndFuseInternal()
Dhlo_instruction.h1473 const std::vector<HloInstruction*>& fused_parameters() const;
Dhlo_instructions.h781 const std::vector<HloInstruction*>& fused_parameters() const;
Dhlo_instruction.cc3346 const std::vector<HloInstruction*>& HloInstruction::fused_parameters() const { in fused_parameters() function in xla::HloInstruction
3347 return Cast<HloFusionInstruction>(this)->fused_parameters(); in fused_parameters()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc34 params->insert(std::end(*params), std::begin(instr.fused_parameters()), in AppendParams()
35 std::end(instr.fused_parameters())); in AppendParams()
Dir_emitter_unnested.cc3573 unnested_hlo->fused_parameters(), [&](const HloInstruction* parameter) { in NumInputsInvolveInOnlyElementwiseOps()
3586 unnested_hlo->fused_parameters(), [&](const HloInstruction* parameter) { in NumInputsWithMoreElementsThan()