def get_input_metadata_impl(self):
start_binding, end_binding = trt_util.get_active_profile_bindings(
self.context)
# This function always uses binding names of the 0th profile.
return trt_util.get_input_metadata_from_engine(self.context.engine,
start_binding,
end_binding)
def get_input_metadata_impl(self):
bindings_per_profile = trt_util.get_bindings_per_profile(
self.context.engine)
# This function always uses binding names of the 0th profile.
return trt_util.get_input_metadata_from_engine(
self.context.engine,
start_binding=0,
end_binding=bindings_per_profile)