saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-25 19:48:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ort_test_utils: skip creating input if it is an initializer (#6544)

Browse source
This commit is contained in:
Chun-Wei Chen 2021-02-05 17:34:08 -08:00 committed by GitHub
parent ccfd90291b
commit 115e16b37b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tools/python/ort_test_dir_utils.py
View file

@ -21,7 +21,7 @@ def _get_numpy_type(model_info, name):
raise ValueError("{} was not found in the model info.".format(name))
def _create_missing_input_data(model_inputs, name_input_map, symbolic_dim_values_map):
def _create_missing_input_data(model_inputs, name_input_map, symbolic_dim_values_map, initializer_set):
"""
Update name_input_map with random input for any missing values in the model inputs.
@ -32,7 +32,10 @@ def _create_missing_input_data(model_inputs, name_input_map, symbolic_dim_values
for input in model_inputs:
if input.name in name_input_map and name_input_map[input.name] is not None:
continue
# skip if the input has already exists in initializer
# models whose ir_version < 4 can have input same as initializer; no need to create input data
if input.name in initializer_set:
continue
input_type = input.type.WhichOneof('value')
if input_type != 'tensor_type':
raise ValueError('Unsupported model. Need to handle input type of {}'.format(input_type))
@ -126,9 +129,10 @@ def create_test_dir(model_path, root_path, test_name,
if not symbolic_dim_values_map:
symbolic_dim_values_map = {}
_create_missing_input_data(model_inputs, name_input_map, symbolic_dim_values_map)
initializer_set = set()
for initializer in onnx.load(model_path).graph.initializer:
initializer_set.add(initializer.name)
_create_missing_input_data(model_inputs, name_input_map, symbolic_dim_values_map, initializer_set)
save_data("input", name_input_map, model_inputs)
# save expected output data if provided. run model to create if not.