Skip to content

Instantly share code, notes, and snippets.

@rmccorm4
Last active May 10, 2020 21:43
Show Gist options
  • Save rmccorm4/a97b4e82b085dae37f9625547e79b96b to your computer and use it in GitHub Desktop.
Save rmccorm4/a97b4e82b085dae37f9625547e79b96b to your computer and use it in GitHub Desktop.
{
'Profile 0':
{
'actual_input_1':
{
'binding_dtype': DataType.FLOAT,
'binding_index': 0,
'binding_name': 'actual_input_1', # context.get_binding_name(binding_index)
'binding_shape': (-1, 3, 224, 224), # context.get_binding_shape(binding_index)
'binding_type': 'INPUT', # engine.binding_is_input(binding_index) == True
'profile': 0,
'profile_shape': [(1, 3, 224, 224), # min
(32, 3, 224, 224), # opt
(32, 3, 224, 224)] # max
},
'output1':
{
'binding_dtype': DataType.FLOAT,
'binding_index': 1,
'binding_name': 'output1', # context.get_binding_name(binding_index)
'binding_shape': (-1, 1000), # context.get_binding_shape(binding_index)
'binding_type': 'OUTPUT', # engine.binding_is_input(binding_index) == False
'profile': 0
}
},
'Profile 1':
{
'actual_input_1 [profile 1]':
{
'binding_dtype': DataType.FLOAT,
'binding_index': 2,
'binding_name': 'actual_input_1 [profile 1]',
'binding_shape': (-1, 3, 224, 224),
'binding_type': 'INPUT',
'profile': 1,
'profile_shape': [(32, 3, 224, 224),
(64, 3, 224, 224),
(64, 3, 224, 224)]
},
'output1 [profile 1]':
{
'binding_dtype': DataType.FLOAT,
'binding_index': 3,
'binding_name': 'output1 [profile 1]',
'binding_shape': (-1, 1000),
'binding_type': 'OUTPUT',
'profile': 1
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment