Searched defs:param_no (Results 1 – 7 of 7) sorted by relevance
51 const ShapeLayout& parameter_layout(int64 param_no) const { in parameter_layout()54 ShapeLayout* mutable_parameter_layout(int64 param_no) { in mutable_parameter_layout()72 const Shape& parameter_shape(int64 param_no) const { in parameter_shape()
79 int64 param_no = instruction->parameter_number(); in HloComputation() local149 int64 param_no, HloInstruction* old_instruction, in ReplaceEntryComputationParameter()168 Status HloComputation::RemoveParameter(int64 param_no) { in RemoveParameter()214 const int64 param_no = i - removed; in RemoveUnusedParametersImpl() local699 int64 param_no = instruction->parameter_number(); in CreateFromProto() local
196 HloInstruction* parameter_instruction(int64 param_no) const { in parameter_instruction()
771 int64 param_no = fused_param->parameter_number(); in HandleFusion() local1693 int64 param_no = fused_param->parameter_number(); in CheckFusionInstruction() local
1451 const int64 param_no = operand_count(); in AddFusionOperand() local
265 void ForceParameterLayout(HloModule* module, int64 param_no, in ForceParameterLayout()
311 int64 param_no = allocation.parameter_number(); in BufferForAllocation() local