Skip to content

Commit

Permalink
Fix TF2 Keras layer tests
Browse files Browse the repository at this point in the history
Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
  • Loading branch information
rkazants committed Aug 30, 2024
1 parent c76e564 commit 94ee557
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions src/frontends/tensorflow/src/input_model.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -271,6 +271,16 @@ void InputModel::InputModelTFImpl::load_places() {
m_graph_iterator->reset();
m_outputs.clear();

if (m_saved_model_input_names) {
// leave only real inputs for the main graph
// if input signature in the input model is defined
m_inputs.clear();
for (const auto& map_name : *m_saved_model_input_names) {
const auto& input_internal_tensor_name = map_name.first;
m_inputs.push_back(m_default_places[input_internal_tensor_name]);
}
}

// SavedModel, MetaGraph formats have model signature that provides a concrete list of outputs
// some output can place among intermediate layers (i.e. it can have its output consumers)
// so just terminal nodes may not cover the whole list of outputs
Expand Down

0 comments on commit 94ee557

Please sign in to comment.