(graph, ntype_id)
| 110 | |
| 111 | |
| 112 | def _get_inner_node_mask(graph, ntype_id): |
| 113 | if NTYPE in graph.ndata: |
| 114 | dtype = F.dtype(graph.ndata["inner_node"]) |
| 115 | return ( |
| 116 | graph.ndata["inner_node"] |
| 117 | * F.astype(graph.ndata[NTYPE] == ntype_id, dtype) |
| 118 | == 1 |
| 119 | ) |
| 120 | else: |
| 121 | return graph.ndata["inner_node"] == 1 |
| 122 | |
| 123 | |
| 124 | def _get_inner_edge_mask(graph, etype_id): |