Unverified Commit ec935c0b authored by Ivan Tikhonov's avatar Ivan Tikhonov Committed by GitHub

Align the number of ti inputs (#4235)

Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
parent 4c322687
...@@ -141,13 +141,6 @@ shared_ptr<op::TensorIterator::OutputDescription> ...@@ -141,13 +141,6 @@ shared_ptr<op::TensorIterator::OutputDescription>
Input<Node> op::TensorIterator::input_for_value(const Output<Node>& value) Input<Node> op::TensorIterator::input_for_value(const Output<Node>& value)
{ {
for (auto input : inputs())
{
if (input.get_source_output() == value)
{
return input;
}
}
auto input_index = get_input_size(); auto input_index = get_input_size();
set_argument(input_index, value); set_argument(input_index, value);
return Input<Node>(this, input_index); return Input<Node>(this, input_index);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment