Searched refs:_GetGraphDef (Results 1 – 2 of 2) sorted by relevance
114 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): member in QuantizationAwareTrainingMNISTTest214 graph_def = self._GetGraphDef(use_trt, batch_size, model_dir)
796 def _GetGraphDef(self, run_params, gdef_or_saved_model_dir): member in TfTrtIntegrationTestBase896 original_gdef = self._GetGraphDef(run_params,898 gdef_to_verify = self._GetGraphDef(run_params,