auto graph_input = _loco_graph->inputs()->create();
graph_input->name(op.getName());
graph_input->dtype(loco::DataType::FLOAT32); // TODO Support other types
- graph_input->node(pull_node);
+ loco::link(graph_input, pull_node);
// Not set inputs
// Add to map
_mir2loco_map.emplace(&op, pull_node);
auto graph_output = _loco_graph->outputs()->create();
graph_output->name(op.getName());
graph_output->dtype(loco::DataType::FLOAT32); // TODO Support other types
- graph_output->node(push_node);
+ loco::link(graph_output, push_node);
// Add to map
_mir2loco_map.emplace(&op, push_node);
}