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()
82 int64_t param_no = instruction->parameter_number(); in HloComputation() local160 int64_t param_no, HloInstruction* old_instruction, in ReplaceEntryComputationParameter()179 Status HloComputation::RemoveParameter(int64_t param_no) { in RemoveParameter()225 const int64_t param_no = i - removed; in RemoveUnusedParametersImpl() local720 int64_t param_no = instruction->parameter_number(); in CreateFromProto() local
202 HloInstruction* parameter_instruction(int64_t param_no) const { in parameter_instruction()
1108 int64_t param_no = fused_param->parameter_number(); in HandleFusion() local2045 int64_t param_no = fused_param->parameter_number(); in CheckFusionInstruction() local
1588 const int64_t param_no = operand_count(); in AddFusionOperand() local
268 void ForceParameterLayout(HloModule* module, int64_t param_no, in ForceParameterLayout()
325 int64_t param_no = allocation.parameter_number(); in BufferForAllocation() local