Python get_input_metadata_from_engine Examples

Example #1

File: runner.py Project: stjordanis/TensorRT

 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)

Example #2

File: runner.py Project: clayne/TensorRT

 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)