Searched defs:input_start_indices (Results 1 – 2 of 2) sorted by relevance
529 HloInstruction* input_start_indices = operands(2); in CreateFromProto() local1237 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermute()1259 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermuteStart()
835 HloInstruction* output, HloInstruction* input_start_indices, in HloCollectivePermuteInstruction()