Exemplos de get_input_metadata_from_engine em Python

Exemplo n.º 1

Arquivo: runner.py Projeto: 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)

Exemplo n.º 2

Arquivo: runner.py Projeto: 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)