graphlearn_torch/python/distributed/dist_neighbor_sampler.py [335:347]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
          if output.edge is not None:
            edge_dict[etype] = output.edge

        if len(nbr_dict) == 0:
          continue
        nodes_dict, rows_dict, cols_dict = inducer.induce_next(nbr_dict)
        merge_dict(nodes_dict, out_nodes)
        merge_dict(rows_dict, out_rows)
        merge_dict(cols_dict, out_cols)
        merge_dict(edge_dict, out_edges)
        count_dict(nodes_dict, num_sampled_nodes, i + 2)
        count_dict(cols_dict, num_sampled_edges, i + 1)
        src_dict = nodes_dict
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



graphlearn_torch/python/sampler/neighbor_sampler.py [277:288]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if output.edge is not None:
              edge_dict[etype] = output.edge
      if len(nbr_dict) == 0:
        continue
      nodes_dict, rows_dict, cols_dict = inducer.induce_next(nbr_dict)
      merge_dict(nodes_dict, out_nodes)
      merge_dict(rows_dict, out_rows)
      merge_dict(cols_dict, out_cols)
      merge_dict(edge_dict, out_edges)
      count_dict(nodes_dict, num_sampled_nodes, i + 2)
      count_dict(cols_dict, num_sampled_edges, i + 1)
      src_dict = nodes_dict
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -