コード例 #1
0
ファイル: runner.py プロジェクト: 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)
コード例 #2
0
ファイル: runner.py プロジェクト: 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)