Home
last modified time | relevance | path

Searched refs:referenced_model_to_index (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/tools/test_generator/
Dtest_generator.py636 def CompileReferencedModels(self, referenced_models, referenced_model_to_index): argument
642 if key not in referenced_model_to_index:
643 referenced_model_to_index[key] = len(referenced_model_to_index)
645 model.Compile(referenced_models, referenced_model_to_index)
646 operand.value = [referenced_model_to_index[key]]
648 def Compile(self, referenced_models=None, referenced_model_to_index=None): argument
654 referenced_model_to_index = {}
659 self.CompileReferencedModels(referenced_models, referenced_model_to_index)