Searched refs:parameter_instructions (Results 1 – 10 of 10) sorted by relevance
82 const std::vector<HloInstruction*>& parameter_instructions = in AddAliasingInformationToIrArray() local83 module_.entry_computation()->parameter_instructions(); in AddAliasingInformationToIrArray()84 if (absl::c_linear_search(parameter_instructions, &hlo)) { in AddAliasingInformationToIrArray()
141 for (auto* param : comp->parameter_instructions()) { in ConvertCalledComputations()222 for (auto* param : comp->parameter_instructions()) { in HandleMultipleOutputs()296 for (auto* param : comp->parameter_instructions()) { in HandleInstruction()
166 const std::vector<HloInstruction*>& parameter_instructions() const { in parameter_instructions() function
1217 fused_instructions_computation()->parameter_instructions().size()); in AddFusionOperand()1348 return fused_instructions_computation()->parameter_instructions(); in fused_parameters()1414 fused_instructions_computation()->parameter_instructions(); in CloneAndFuseInternal()1443 fused_instructions_computation()->parameter_instructions(); in CloneAndFuseInternal()
348 for (auto* param : entry->parameter_instructions()) { in AddCopiesForAliasedInputOutputs()
53 nested_computation.parameter_instructions()) { in EmitBasePointersForNestedComputation()
267 for (const auto& param : comp->parameter_instructions()) { in DoInfoCommand()
78 for (const auto* param : entry->parameter_instructions()) { in GetProgramShapeWithLayout()
558 const auto params = module->entry_computation()->parameter_instructions(); in MakeFakeArguments()
128 for (const HloInstruction* param : computation->parameter_instructions()) { in EmitComputation()