Searched refs:gdef_or_saved_model_dir (Results 1 – 1 of 1) sorted by relevance
796 def _GetGraphDef(self, run_params, gdef_or_saved_model_dir): argument797 if isinstance(gdef_or_saved_model_dir, str):799 root = load.load(gdef_or_saved_model_dir)812 gdef_or_saved_model_dir, tag_constants.SERVING).graph_def813 assert isinstance(gdef_or_saved_model_dir, graph_pb2.GraphDef)814 return gdef_or_saved_model_dir