Searched refs:visiting_hlo (Results 1 – 3 of 3) sorted by relevance
85 HloInstruction* visiting_hlo() const { return visiting_hlo_; } in visiting_hlo() function
909 HloInstruction* visiting_hlo = hlo; in ExchangeHalo() local911 auto concat = ExchangeHalo(visiting_hlo, left_halo_size_functions[dim], in ExchangeHalo()917 visiting_hlo = *concat; in ExchangeHalo()919 return visiting_hlo; in ExchangeHalo()
773 HloInstruction* visiting_hlo = hlo_; in ReshardAsWindowedInput() local805 visiting_hlo, base_shape_, left_halo_size_functions[dim], in ReshardAsWindowedInput()816 visiting_hlo = *resharded; in ReshardAsWindowedInput()819 visiting_hlo, shard_window, in ReshardAsWindowedInput()