Searched defs:param_no (Results 1 – 7 of 7) sorted by relevance
51 const ShapeLayout& parameter_layout(int64_t param_no) const { in parameter_layout()54 ShapeLayout* mutable_parameter_layout(int64_t param_no) { in mutable_parameter_layout()72 const Shape& parameter_shape(int64_t param_no) const { in parameter_shape()
90 int64_t param_no = instruction->parameter_number(); in HloComputation() local183 int64_t param_no, HloInstruction* old_instruction, in ReplaceEntryComputationParameter()202 Status HloComputation::RemoveParameter(int64_t param_no) { in RemoveParameter()226 int64_t param_no, std::unique_ptr<HloInstruction> instruction) { in ReplaceParameter()265 const int64_t param_no = i - removed; in RemoveUnusedParametersImpl() local692 int64_t param_no = instruction->parameter_number(); in CreateFromProto() local
223 HloInstruction* parameter_instruction(int64_t param_no) const { in parameter_instruction()
1141 int64_t param_no = fused_param->parameter_number(); in HandleFusion() local2265 int64_t param_no = fused_param->parameter_number(); in CheckFusionInstruction() local
1599 const int64_t param_no = operand_count(); in AddCallOperand() local
300 void ForceParameterLayout(HloModule* module, int64_t param_no, in ForceParameterLayout()
466 int64_t param_no = allocation.parameter_number(); in BufferForAllocation() local