Searched refs:_GetConfigProto (Results 1 – 2 of 2) sorted by relevance
101 def _GetConfigProto(self): member in TrtConvertTest138 with self.session(graph=g, config=self._GetConfigProto()) as sess:162 with self.session(graph=g, config=self._GetConfigProto()) as sess:184 session_config=self._GetConfigProto(),272 with self.session(config=self._GetConfigProto()) as sess:389 with self.session(config=self._GetConfigProto()) as sess:400 with self.session(config=self._GetConfigProto()) as sess:427 with self.session(config=self._GetConfigProto()) as sess:445 with self.session(config=self._GetConfigProto()) as sess:
217 def _GetConfigProto(self, run_params, graph_state): member in TfTrtIntegrationTestBase327 session_config = self._GetConfigProto(run_params, GraphState.CALIBRATE)344 session_config = self._GetConfigProto(run_params, GraphState.INFERENCE)508 config_no_trt = self._GetConfigProto(run_params, GraphState.ORIGINAL)527 infer_config = self._GetConfigProto(run_params, GraphState.INFERENCE)