Searched refs:_GetConfigProto (Results 1 – 2 of 2) sorted by relevance
81 def _GetConfigProto(self, rewriter_config=None): member in TrtConvertTest139 with self.session(graph=g, config=self._GetConfigProto()) as sess:169 with self.session(graph=g, config=self._GetConfigProto()) as sess:311 with self.session(config=self._GetConfigProto()) as sess:355 config = self._GetConfigProto(795 with self.session(config=self._GetConfigProto()) as sess:806 with self.session(config=self._GetConfigProto()) as sess:828 with self.session(config=self._GetConfigProto()) as sess:838 with self.session(config=self._GetConfigProto()) as sess:
332 def _GetConfigProto(self, run_params, graph_state): member in TfTrtIntegrationTestBase448 config = self._GetConfigProto(run_params, GraphState.INFERENCE)946 graph=g, config=self._GetConfigProto(run_params,