Searched defs:leaf_index (Results 1 – 5 of 5) sorted by relevance
134 HloComputation* comp) { in ConvertType()190 HloComputation* comp) { in ChangeOutputTypeThenInsertConvertBack()
759 HloComputation* comp) { in ResolveInconsistentFusions()858 HloComputation* comp) { in Run()923 HloComputation* comp) { in Run()
776 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index, in DeepCopyHelper()845 HloInstruction*(HloInstruction* leaf, const ShapeIndex& leaf_index, in DeepCopyInstructionWithCustomCopier()
349 int leaf_index = 0; in ComputeHloReplication() local
353 int *(*leaf_index)(struct tree *, void *); member