Searched defs:input_start_indices (Results 1 – 2 of 2) sorted by relevance
585 HloInstruction* input_start_indices = operands(2); in CreateFromProto() local1334 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermute()1356 HloInstruction* input_start_indices, HloInstruction* output_start_indices, in CreateCollectivePermuteStart()
999 HloInstruction* output, HloInstruction* input_start_indices, in HloCollectivePermuteInstruction()