Searched refs:_GetGraphDef (Results 1 – 2 of 2) sorted by relevance
113 def _GetGraphDef(self, use_trt, max_batch_size, model_dir): member in QuantizationAwareTrainingMNISTTest219 graph_def = self._GetGraphDef(use_trt, batch_size, model_dir)
135 def _GetGraphDef(self): member in TrtConvertTest182 input_graph_def=None if input_saved_model_dir else self._GetGraphDef(),