Removed propagate_rt_info function
This commit is contained in:
parent
60c74a2879
commit
514a21b93c
@ -1225,35 +1225,6 @@ bool could_propagate(const Output<Node>& output, std::vector<Node*>& order)
|
||||
return status;
|
||||
}
|
||||
|
||||
void propagate_rt_info(Node* node, const Output<Node>& final_port)
|
||||
{
|
||||
auto node_outputs = node->outputs();
|
||||
bool same_outputs =
|
||||
std::all_of(node_outputs.begin(), node_outputs.end(), [](const Output<Node>& output) {
|
||||
return output.get_tensor().has_and_set_bound();
|
||||
});
|
||||
if (same_outputs && op::is_constant(node)) // constant should not propagate it's rt_info
|
||||
{
|
||||
std::unordered_set<Node*> stop_nodes;
|
||||
for (const auto& in : final_port.get_target_inputs())
|
||||
stop_nodes.insert(in.get_node());
|
||||
|
||||
auto curr_node = node->shared_from_this();
|
||||
for (const auto& output : node_outputs)
|
||||
{
|
||||
if (output == final_port)
|
||||
continue;
|
||||
for (auto& in : output.get_target_inputs())
|
||||
{
|
||||
if (stop_nodes.count(in.get_node()))
|
||||
continue;
|
||||
auto consumer = in.get_node()->shared_from_this();
|
||||
copy_runtime_info({curr_node, consumer}, consumer);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
HostTensorPtr evaluate_bound(const Output<Node>& output, bool is_upper)
|
||||
{
|
||||
// bound is already set in the tensor
|
||||
@ -1291,7 +1262,6 @@ HostTensorPtr evaluate_bound(const Output<Node>& output, bool is_upper)
|
||||
for (const auto& input : input_values)
|
||||
if (input.get_target_inputs().size() == 1)
|
||||
input.get_tensor().invalidate_values();
|
||||
propagate_rt_info(node, output);
|
||||
}
|
||||
else
|
||||
{
|
||||
|
Loading…
Reference in New Issue
Block a user