Searched refs:parameter_instructions (Results 1 – 15 of 15) sorted by relevance
203 std::vector<HloInstruction*> parameter_instructions; in IsAllReduceMovable() local205 std::back_inserter(parameter_instructions), in IsAllReduceMovable()216 for (HloInstruction* parameter_instruction : parameter_instructions) { in IsAllReduceMovable()
252 for (auto* param : comp->parameter_instructions()) { in ConvertCalledComputations()334 for (auto* param : comp->parameter_instructions()) { in HandleMultipleOutputs()407 for (auto* param : comp->parameter_instructions()) { in HandleInstruction()
203 const std::vector<HloInstruction*>& parameter_instructions() const { in parameter_instructions() function
1450 fused_instructions_computation()->parameter_instructions().size()); in AddFusionOperand()1617 return fused_instructions_computation()->parameter_instructions(); in fused_parameters()1683 fused_instructions_computation()->parameter_instructions(); in CloneAndFuseInternal()1712 fused_instructions_computation()->parameter_instructions(); in CloneAndFuseInternal()
360 for (auto* param : entry->parameter_instructions()) { in AddCopiesForAliasedInputOutputs()
847 called_computation->parameter_instructions()) { in FindAliases()1728 entry_computation->parameter_instructions()) { in AddInputAndOutputRequiredAssignments()
1697 for (auto* param : sub_computation->parameter_instructions()) { in DefaultAction()
111 module->entry_computation()->parameter_instructions()) { in AssignMemorySpace()181 entry_computation->parameter_instructions()) { in CheckParametersInDefaultMemory()
61 nested_computation_.parameter_instructions()) { in CodegenNestedComputation()
293 for (const auto& param : comp->parameter_instructions()) { in DoInfoCommand()
129 TF_RETURN_IF_ERROR(GetMlirTypes(computation.parameter_instructions(), &args)); in ImportAsFunc()163 TF_RETURN_IF_ERROR(GetMlirTypes(computation.parameter_instructions(), &args)); in ImportAsRegion()
78 for (const auto* param : entry->parameter_instructions()) { in GetProgramShapeWithLayout()
685 const auto params = module->entry_computation()->parameter_instructions(); in MakeFakeArguments()
173 for (const HloInstruction* param : computation->parameter_instructions()) { in EmitComputation()
3788 for (const HloInstruction* param : entry->parameter_instructions()) { in PreprocessSharding()