Searched refs:new_node_map (Results 1 – 5 of 5) sorted by relevance
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/pass/ |
| H A D | place_device.cc | 109 std::vector<NodePtr> new_node_map(idx.num_nodes(), nullptr); in PlaceDevice() local
130 if (new_node_map[e.node_id] != nullptr || dev_id != device[e.node_id]) { in PlaceDevice()
136 if (new_node_map[cid] != nullptr) { in PlaceDevice()
162 if (new_node_map[e.node_id] != nullptr) { in PlaceDevice()
164 NodeEntry{new_node_map[e.node_id], e.index, 0}); in PlaceDevice()
177 if (new_node_map[e.node_id] != nullptr) { in PlaceDevice()
179 NodeEntry{new_node_map[e.node_id], e.index, 0}); in PlaceDevice()
188 if (new_node_map[cid] != nullptr) { in PlaceDevice()
189 new_node->control_deps.push_back(new_node_map[cid]); in PlaceDevice()
195 new_node_map[nid] = std::move(new_node); in PlaceDevice()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/nnvm/src/pass/ |
| H A D | place_device.cc | 108 std::vector<ObjectPtr> new_node_map(idx.num_nodes(), nullptr); in PlaceDevice() local
121 if (new_node_map[e.node_id] != nullptr || dev_id != device[e.node_id]) { in PlaceDevice()
128 if (new_node_map[e.node_id] != nullptr || dev_id != device[e.node_id]) { in PlaceDevice()
135 if (new_node_map[cid] != nullptr) { in PlaceDevice()
161 if (new_node_map[e.node_id] != nullptr) { in PlaceDevice()
174 if (new_node_map[e.node_id] != nullptr) { in PlaceDevice()
184 if (new_node_map[cid] != nullptr) { in PlaceDevice()
185 new_node->control_deps.push_back(new_node_map[cid]); in PlaceDevice()
191 new_node_map[nid] = std::move(new_node); in PlaceDevice()
199 if (new_node_map[idx.node_id(e.node.get())] != nullptr) { in PlaceDevice()
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/pass/ |
| H A D | place_device.cc | 109 std::vector<NodePtr> new_node_map(idx.num_nodes(), nullptr); in PlaceDevice() local
130 if (new_node_map[e.node_id] != nullptr || dev_id != device[e.node_id]) { in PlaceDevice()
136 if (new_node_map[cid] != nullptr) { in PlaceDevice()
162 if (new_node_map[e.node_id] != nullptr) { in PlaceDevice()
164 NodeEntry{new_node_map[e.node_id], e.index, 0}); in PlaceDevice()
177 if (new_node_map[e.node_id] != nullptr) { in PlaceDevice()
179 NodeEntry{new_node_map[e.node_id], e.index, 0}); in PlaceDevice()
188 if (new_node_map[cid] != nullptr) { in PlaceDevice()
189 new_node->control_deps.push_back(new_node_map[cid]); in PlaceDevice()
195 new_node_map[nid] = std::move(new_node); in PlaceDevice()
[all …]
|
| /dports/net/mpich/mpich-3.4.3/src/mpid/ch4/src/ |
| H A D | ch4r_proc.c | 98 int *new_node_map = NULL; in alloc_globals_for_avtid() local
102 new_node_map = (int *) MPL_malloc(MPIDI_av_table[avtid]->size * sizeof(int), MPL_MEM_ADDRESS); in alloc_globals_for_avtid()
103 MPIR_ERR_CHKANDJUMP(new_node_map == NULL, mpi_errno, MPI_ERR_NO_MEM, "**nomem"); in alloc_globals_for_avtid()
104 MPIDI_global.node_map[avtid] = new_node_map; in alloc_globals_for_avtid()
|
| /dports/net-p2p/c-lightning/lightning-0.10.2/gossipd/ |
| H A D | routing.c | 115 static struct node_map *new_node_map(const tal_t *ctx) in new_node_map() function
286 rstate->nodes = new_node_map(rstate); in new_routing_state()
|