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()
159 HloInstruction* parameter_instruction(int64 param_no) const { in parameter_instruction()
79 int64 param_no = instruction->parameter_number(); in HloComputation() local145 Status HloComputation::RemoveParameter(int64 param_no) { in RemoveParameter()182 const int64 param_no = i - removed; in RemoveUnusedParameters() local571 int64 param_no = instruction->parameter_number(); in CreateFromProto() local
523 int64 param_no = fused_param->parameter_number(); in HandleFusion() local1256 int64 param_no = fused_param->parameter_number(); in CheckFusionInstruction() local
1218 const int64 param_no = operand_count(); in AddFusionOperand() local
264 void ForceParameterLayout(HloModule* module, int64 param_no, in ForceParameterLayout()
255 auto param_no = allocation.parameter_number(); in ExecuteOnStream() local