Searched refs:i_neighs (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_bgcf.py | 138 i_neighs, argument
156 i_neigh_matrix_at_layers = self.gather(self.user_embed, i_neighs, 0)
188 def construct(self, users, items, neg_items, u_neighs, u_gnew_neighs, i_neighs, i_gnew_neighs): argument
190 … u_neighs, u_gnew_neighs, items, i_neighs, i_gnew_neighs,
191 items, i_neighs, i_gnew_neighs, 1)
|